Fix the test demo app