Fixing test cases for travis