Removing import statement breaking travis