corrected return bug