Minor bug fix with message output.