Bug fix: missing numpy import