Correct some imports for the travis build