add missing imports
Merge pull request #454 from racker/fixes/missing_imports

add missing imports