Fix code consistency