remove bug that made the unit-test break