Changed test a little bit