Added a return

Added a return