Todo fixes