trying to fix up the test