Add missing import
Attempt to get travis to run