Accidentally removed numpy import in previous commit