fix missing import

fixing missing import of previously committed fix for ERRAI-470