Fixed test after renaming function.