Fixed issue in demo project