Fixes TODO comment