fix final todo