Fix 2 TODO items: