Changed test after feedback