Fix test from failing due to IP