Update sanity test to demo