minor additoin to TODO