Small code oversight fixed