diff --git a/paddleslim/nas/sa_nas.py b/paddleslim/nas/sa_nas.py
index 00decbfd1ae38dfa3fedf3234665ca740674d603..1563e141b098fb59e6b2f5abf6c4b5f8cd92d275 100644
--- a/paddleslim/nas/sa_nas.py
+++ b/paddleslim/nas/sa_nas.py
@@ -102,7 +102,7 @@ class SANAS(object):
         return socket.gethostbyname(socket.gethostname())
 
     def tokens2arch(self, tokens):
-        return self._search_space.token2arch(self.tokens)
+        return self._search_space.token2arch(tokens)
 
     def next_archs(self):
         """