Fixed unit test compilation