Fixed unit test check