small bugfix, missing import