Fixed Numpy import that was missin.