From a33202f0dcb1df6bb65b2e4b69529c2f6e5d20f7 Mon Sep 17 00:00:00 2001
From: rkdavid <rkdavid@apstra.com>
Date: Mon, 5 Sep 2022 19:56:00 +0200
Subject: [PATCH] Fix Issue#36

---
 .gitignore     | 1 +
 aos/devices.py | 4 ++--
 setup.py       | 2 +-
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/.gitignore b/.gitignore
index a9e6d3c..7505967 100644
--- a/.gitignore
+++ b/.gitignore
@@ -8,6 +8,7 @@
 .pydevproject
 .idea
 aos_api_client.egg-info
+apstra_api_python.egg-info
 build
 dist
 */.pytest_cache
diff --git a/aos/devices.py b/aos/devices.py
index 8245558..7446f6e 100644
--- a/aos/devices.py
+++ b/aos/devices.py
@@ -183,7 +183,7 @@ def find_system_with_ip(self, ip_addr: str) -> Optional[System]:
         for system in self.iter_all():
             if system.facts["mgmt_ipaddr"] == ip_addr:
                 return system
-            return NullSystem
+        return NullSystem
 
     def delete(self, agent_uuid: str) -> None:
         self.rest.delete(f"/api/system-agents/{agent_uuid}")
@@ -220,7 +220,7 @@ def find_agent_with_ip(self, ip_addr: str) -> Optional[SystemAgent]:
         for agent in self.iter_all():
             if agent.management_ip == ip_addr:
                 return agent
-            return NullSystemAgent
+        return NullSystemAgent
 
     def get_packages(self):
         """
diff --git a/setup.py b/setup.py
index 673ed59..05719e4 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@
 
 NAME = "apstra-api-python"
 
-VERSION = '0.1.19'
+VERSION = '0.1.20'
 
 
 REQUIRES = (["requests==2.24.0"],)