demo module bug fix