Fixed unit test