fixing broken unit test from prev commit