Changed test a bit