TODO tweaks