Updated unit tests to check the IP address