diff --git a/test/test_model.jl b/test/test_model.jl
index 99422a9..a8f3acc 100644
--- a/test/test_model.jl
+++ b/test/test_model.jl
@@ -86,8 +86,8 @@ Dict()
 """ ->
 function get_nodes(model, node_ids=[]; partition=1, time=0, increment=0)
   subset = Dict()
-  all_node_coords = h5read("$model.1.h5", "$time/$increment/nodes/coords")
-  all_node_ids = h5read("$model.1.h5", "$time/$increment/nodes/node_ids")
+  all_node_coords = h5read("$model.$partition.h5", "$time/$increment/nodes/coords")
+  all_node_ids = h5read("$model.$partition.h5", "$time/$increment/nodes/node_ids")
   @debug("all node coords: ", all_node_coords)
   @debug("all node ids: ", all_node_ids)
   if length(node_ids) == 0
@@ -132,4 +132,4 @@ facts("get nodes from model") do
   subset = get_nodes(model, [1,2])
   @fact subset[1] => [1.0, 2.0, 3.0]
   @fact subset[2] => [2.0, 3.0, 4.0]
-end
\ No newline at end of file
+end