Forgot to update url import in tests