Minor bug fix (not really)