Add missing imports

This should fix bug #939 http://fiji.sc/bugzilla/show_bug.cgi?id=939