fix 2 TODOs