diff --git a/imperative/python/megengine/functional/__init__.py b/imperative/python/megengine/functional/__init__.py
index c2a78a462574547911ff1abd283b91b4fbd81822..86e1ea25878cb5bb98a734338135dfc8d1def096 100644
--- a/imperative/python/megengine/functional/__init__.py
+++ b/imperative/python/megengine/functional/__init__.py
@@ -1,13 +1,11 @@
 # -*- coding: utf-8 -*-
 # pylint: disable=redefined-builtin
-from . import metric, utils, vision
 from .elemwise import *
 from .math import *
 from .nn import *
 from .tensor import *
-from .utils import *
 
-from . import distributed  # isort:skip
+from . import utils, vision, distributed  # isort:skip
 
 # delete namespace
 # pylint: disable=undefined-variable
diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py
index f4b998e349c62ccfb9179e4c220d4bcc4b4b18aa..29517e1254a08bc518e5be31b07778b92968949a 100644
--- a/imperative/python/megengine/functional/nn.py
+++ b/imperative/python/megengine/functional/nn.py
@@ -43,6 +43,7 @@ from .debug_param import get_execution_strategy
 from .distributed import all_reduce_sum
 from .elemwise import _elwise, exp, log, log1p, maximum, minimum
 from .math import max, sum
+from .metric import topk_accuracy
 from .tensor import broadcast_to, concat, expand_dims, ones, squeeze, zeros
 
 __all__ = [
@@ -85,6 +86,7 @@ __all__ = [
     "softmax",
     "softplus",
     "sync_batch_norm",
+    "topk_accuracy",
     "warp_affine",
     "warp_perspective",
     "pixel_shuffle",
@@ -1960,5 +1962,4 @@ def pixel_shuffle(inp: Tensor, upscale_factor: int) -> Tensor:
 
 from .quantized import conv_bias_activation  # isort:skip
 from .loss import *  # isort:skip
-from .metric import *  # isort:skip
 from .vision import *  # isort:skip
diff --git a/imperative/python/megengine/functional/utils.py b/imperative/python/megengine/functional/utils.py
index b24272d0a1c4d3f570df158efd83e563c3a8d327..5b337fb4f0067c02285cff83c2976ee625926c8b 100644
--- a/imperative/python/megengine/functional/utils.py
+++ b/imperative/python/megengine/functional/utils.py
@@ -7,8 +7,6 @@ from ..utils.deprecation import deprecated_func
 from .elemwise import abs, maximum, minimum
 from .tensor import ones, zeros
 
-__all__ = ["topk_accuracy"]
-
 
 def _assert_equal(
     expect: Tensor, actual: Tensor, *, maxerr: float = 0.0001, verbose: bool = False