Revert "changes on test classs"

This reverts commit 1630ff45a8692bae590d66309fe6c610d0f489b3.