Minor bug fix and update.