Fixing test for travis