fixes #1 in demo