fix missing import
Merge pull request #34 from pslegr/ERRAI-470_1

fix missing import