Fix test to support pandas