Update the umd callback right away.

This fixes umd/callbacks/umd.