fixed last TODO