Fixed inconsistency in code