Changed bad import

I messed this up. I would assume it's not breaking because of a .pyc file hanging around or something.

All tests pass.