fix unit test broken in previous commit