Fixed a missing import that TravisCI needs.