main: fix TODO