Fixed an ImportError that somehow sneaked in.
Bumped version to 0.2.1