Missing import
Simplify return from new mode call