Calculations & Assign Embedded Data Working in Preview Mode but not Distributions?

Hi everyone! I have some javascript code that I've written that takes survey question responses, calculates a score, and assigns a segment name as embedded data. It all works fine in preview mode - once survey responses are submitted, I can see that the embedded data segment field is filled out. However, when I tested this with a distribution (emailing myself), the embedded data field doesn't seem to get assigned, and shows up blank in my data. Any ideas why this might be the case? Thanks!