Add missing import statements

In your recent merge, a few import statement went AWOL, meaning the plugin was completely broken.