Change function definition for consistency