Small bugfix

Small bugfix