Correcting a return value