diff --git a/policyopt/nn.py b/policyopt/nn.py
index 64a0d5c..47755f3 100644
--- a/policyopt/nn.py
+++ b/policyopt/nn.py
@@ -390,7 +390,7 @@ def write_snapshot(self, model, key_iter):
             self.f.create_array(groupname, arrayname, v.get_value(), createparents=True)
 
         # Store the model hash as an attribute
-        self.f.getNode(snapshot_root)._v_attrs.hash = model.savehash()
+        self.f.get_node(snapshot_root)._v_attrs.hash = model.savehash()
 
         self.f.flush()