Revert "minor code improvement"

This reverts commit 6e92b2fd193d52232fbede028f9705deccc388e2.
fix minor code "improvement"