[BUGFIX] Add missing import