fix TODOs