Numpy error fixed