Fix tests with numpy expressions