Carlos Correa talking with Astros about potential return