Fixed test thanks to @grumpydev comments