Revert "delete code for test working"

This reverts commit 87d423c1b27fb98ae3d7223548dd99181ffe8837.