You probably have to change the logic in the script a bit. I don't know offhand but I think on a beforesubmit you save the record differently to aftersubmit