Quick fix, demo update