fixed bug with numpy