Fixed return statement/function