fix tests based on demo data updates