#95 unit test fix