fixed error with return