BUG: Bug fix with output message implementation