Add missing package import