updated test TODO