Fixed numpy import