Fixed an import
Fix the travis config.