change  import for Travis