diff --git a/pyproject.toml b/pyproject.toml
index a1573f2c6..ccb1287c7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -164,3 +164,5 @@ exclude = [".git", ".github", ".venv", "build", "docs", "examples", "test"]
 line-length = 92
 [tool.ruff.lint.per-file-ignores]
 "__init__.py" = ["E402", "F401", "F403"]
+[tool.ruff.lint]
+extend-ignore = ["F401"]
diff --git a/src/porespy/generators/_pseudo_packings.py b/src/porespy/generators/_pseudo_packings.py
index d2aaf7935..f1bdcc2e4 100644
--- a/src/porespy/generators/_pseudo_packings.py
+++ b/src/porespy/generators/_pseudo_packings.py
@@ -210,7 +210,7 @@ def pseudo_gravity_packing(
     # Generate elevation values to initialize queue
     from porespy.generators import ramp
     h = ramp(im.shape, inlet=0, outlet=im.shape[0], axis=0)*mask
-    h = h*1.0 + np.random.rand(len(h))
+    h = h*1.0 + np.random.rand(*im.shape)
     tmp = np.arange(h.size)[mask.flatten()]
     inds = np.vstack(np.unravel_index(tmp, im.shape)).T
     h = h.flatten()[mask.flatten()]