Small bugfix

I actually tested it this time ;)