cleaned up test a bit