Fix demo test