Fixed test after nick previous fix