Small bugfix to previous commit