Fixed return variable usage