Small bug fix and update.