Fix numpy import statement