fixed up the first test a bit more