Add missing namespace import