add missing class for optimize control