TODO Fixes