small cleanup
Revert bad change.