Mild TODO cleanup