Fix missing imports
Merge pull request #35 from enthought/fix/missing-imports

Fix missing imports