Modified test to remove matplotlib related color strings