attempt unit test fix