ValueError came from lack of np.squeeze(duration)