This should fix test failing