Updated code for consistency