Minor bugfix