fixing test for demo