fixed error in return