include correct class on production