Little bugfix (hopefully)