Update demo. Fix issue where ejecting would cause an error sometimes.