TODO update

Forgot to mark issue as fixed.