fixing broken unit test