Revert "add the code only when the test run"

This reverts commit dff3b8e9d25a3b6cc3021f43ecd25ba326a481e2.