Fixing missing import for utils