really fix unit test this time