Pass in the element directly instead of passing in "that"