Fix unit test for real