Fixed problem with return