fixing bug for demo