Small bugfix. I need more coffee :)