return statement fixed