fix return from transform