Add missing require
Follow variable name change