Fix typing after formatting in ui demo