fix code demo