Minor bugfix

Minor bugfix