diff --git a/src/BasisConvolution/util/augment.py b/src/BasisConvolution/util/augment.py
index 058bd10..3e8b5f8 100644
--- a/src/BasisConvolution/util/augment.py
+++ b/src/BasisConvolution/util/augment.py
@@ -82,7 +82,6 @@ def loadAugmentedFrame(index, dataset, hyperParameterDict, unrollLength = 8):
 
     # config['neighborhood']['verletScale'] = 1.0
     # config['neighborhood']['scheme'] = 'compact'
-
     searchNeighbors(augmentedStates[0], config, computeKernels = True)
 
     currentState = augmentedStates[0]
diff --git a/src/BasisConvolution/util/testcases.py b/src/BasisConvolution/util/testcases.py
index 4fffea3..1575e88 100644
--- a/src/BasisConvolution/util/testcases.py
+++ b/src/BasisConvolution/util/testcases.py
@@ -420,6 +420,8 @@ def parseSPHConfig(inFile, device, dtype):
     if hasDiffSPH:
         parseDefaultParameters(config)
         parseModuleParameters(config)
+    else:
+        raise ModuleNotFoundError('diffSPH is not installed, cannot parse SPH config')
     
     return config