Little fix on demo