-

prody · Mar 11, 2021 · e9aad40 · e9aad40
1 parent e255fd2
commit e9aad40
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/prody/trajectory/trajbase.py b/prody/trajectory/trajbase.py
@@ -2,7 +2,7 @@
 """This module defines base class for trajectory handling."""
 
 from numbers import Integral
-from numpy import ndarray, unique
+from numpy import ndarray, unique, array
 
 from prody.ensemble import Ensemble
 from prody.utilities import checkCoords, checkWeights
@@ -370,7 +370,7 @@ def hasUnitcell(self):
         pass
 
 
-    def wrap(self, unitcell=None, center=np.array([0., 0., 0.])):
+    def wrap(self, unitcell=None, center=array([0., 0., 0.])):
         """Wrap atoms into an image of the system simulated under periodic boundary
         conditions for all frames and returns a new Trajectory.