fix small todos