Fixed demo dataset bug.