Add missing #import