Fix build
Update TODO