add missing import, unit tests are now fixed