Minor bug fix.