added missing imports
0.6.3