Small fix to context assignment in .on()