fixing demo tests