Added missing import

This import seems to be required on some versions of numpy.
PEP8
Merge pull request #1333 from delallea/minor

Fixed import error with some numpy version(s)