Minor bug fix...