little bug fixed for demo