minor
fix TODO