Fixed unit test mistake