Tiny refactor in views/input-diagram