Changed test a little.