Revert "delete unnecessary class"

This reverts commit b0608971cc2fb219c4b2430749ed0d97158a929a.