attempt to fix unit test