Possible fix to get the test working for CI