Fixing travis tests