Fix contextlib2 import

I've been using your awesome fork, but just noticed this remaining compatibility issue.