Fix demo (Fix #6)