Minor bug fix, Thanks trismarck