Fixes demo script