Revert "small logic fix"

This reverts commit 3f0ba4e440769233714a61a208b98002dca4e094.