Fixed missing numpy import.