TODO fixes.
Whoops.