Revert faulty changes to smoothstate demo