Small bugfix

Bugfix