Add missing tf dependency