Rethink the unit tests on the demo.
- Make sure the demo is only for the demo.
- Intent to move unit test to a more appropriate place.