Add missing import and bracket