minor fix for docs and demo