Changed function signature for consistency