The node you are looking for is at
set-display-callback
.