Accidentally broke snapshotting to UI.
I really need tests for this...