fixing the one unit test I have