Fix matplotlib import in Crab test