fix an error in the return statement