Code consistency fix