diff --git a/nerfies/configs.py b/nerfies/configs.py
index 079689e..50da41a 100644
--- a/nerfies/configs.py
+++ b/nerfies/configs.py
@@ -16,7 +16,7 @@
 from typing import Any, Mapping, Optional, Tuple
 
 import dataclasses
-from flax.linen import nn
+import flax.linen as nn
 import gin
 import immutabledict