Revert "change the class"

This reverts commit d316e629f67c28c914370ef7af7c224cc5af9315.