I just found this has potential risk of failing to get the element if the document is not loaded yet. Could you move this line to `init`?