fix travis ci build
Tests now pass