Small update bugfix