Fixed test which wasn't properly merged