fixed messed up return statement