Fixed error due to numpy version