updated tests to work with demo