fixed class declaration
Revert "fixed class declaration"

This reverts commit 7e146f3b30faab57b2e0079989d7c3f34912d947.