diff --git a/dnn/test/common/topk.h b/dnn/test/common/topk.h
index 0bd5baeb..435759cd 100644
--- a/dnn/test/common/topk.h
+++ b/dnn/test/common/topk.h
@@ -16,7 +16,7 @@ private:
 public:
     OprProxy() = default;
     OprProxy(int k) : m_k{k} {}
-    void init(TopK*, const TensorLayoutArray&) {}
+    void init(TopK*, const TensorNDArray&) {}
 
     void deduce_layout(TopK* opr, TensorLayoutArray& layouts) {
         if (layouts.size() == 3) {