Fix for test failing in travis