Attempt to fix test failure

Attempt to fix test failure