Fixing tests

fixing tests to fix build in travis