todos: Fix todos in diagram listeners