Revert "code fix"

This reverts commit 8c812dafc94747662b78b2e396ba47f24ee315e7.