Correct document.on behavior

document.on now returns document