Chat: Small fix for latest code addition