we shouldnt return ai in forwardProp, so I removed it
merge