attempt to fix unit test problem