Fix on the fly updates again

The last commit was broken. Whoops!