fix reverted code in prior commit