Fixed unit test assertion