Small fix for updating