fixing up a test