Fix for update bug

Hi Felix, just figuring out how to use github so I could give you my fix for that bug.