Fix test broken due to colorama