Fixed error in unit test