fix TODO bug