Fixed imports in travis CI.