Minor fixes

This is really great! I didn't know they provided the `mock` module.

Nitpicks - Can you remove the empty line between imports? And sort them alphabetically? :)