Revert "Revert "test class""

This reverts commit 5f0e2bcdb133feb9d86d6df3173c0601cdc2f41a.