Add missing "use" statement