Minor todo update