Fix for new numpy