Fix 1 unit test