Fixed a subtle ImportError

This will fix Jenkins.