Followup 96ff968: use this.$element after it's initialized, not before

Bug: 67437
Change-Id: I523b773f44915430653ceaf35db0eb5dc0daf019
Merge "Followup 96ff968: use this.$element after it's initialized, not before"