finally fixed todo!