Update test for demo