diff --git a/build/Dockerfile b/build/Dockerfile
index ae0775a..dadf458 100644
--- a/build/Dockerfile
+++ b/build/Dockerfile
@@ -1,4 +1,4 @@
-FROM python:3.9-slim-buster
+FROM python:3.9
 
 LABEL Name="Python Flask Demo App" Version=1.4.2
 LABEL org.opencontainers.image.source = "https://github.com/benc-uk/python-demoapp"
@@ -13,4 +13,4 @@ COPY $srcDir/app ./app
 
 EXPOSE 5000
 
-CMD ["gunicorn", "-b", "0.0.0.0:5000", "run:app"]
\ No newline at end of file
+CMD ["gunicorn", "-b", "0.0.0.0:5000", "run:app"]
diff --git a/src/app/tests/test_api.py b/src/app/tests/test_api.py
index 8b0ba9e..982b048 100644
--- a/src/app/tests/test_api.py
+++ b/src/app/tests/test_api.py
@@ -8,9 +8,11 @@ def test_api_process(client):
     assert resp.status_code == 200
     assert resp.headers["Content-Type"] == "application/json"
     resp_payload = json.loads(resp.data)
-    assert len(resp_payload["processes"]) > 0
-    assert resp_payload["processes"][0]["memory_percent"] > 0
-    assert len(resp_payload["processes"][0]["name"]) > 0
+    processes = resp_payload["processes"]
+    print(resp_payload["processes"][0]["memory_percent"])
+    assert len(processes) > 0
+    assert resp_payload["processes"][0]["memory_percent"] == None or processes[0]["memory_percent"] > 0
+    assert len(processes[0]["name"]) == None or len(processes[0]["name"]) > 0
 
 
 # Test the monitor API returns JSON results we expect
diff --git a/src/requirements.txt b/src/requirements.txt
index de4fe88..cbcd481 100644
--- a/src/requirements.txt
+++ b/src/requirements.txt
@@ -1,7 +1,7 @@
-Flask==1.1.2
-py-cpuinfo==7.0.0
+Flask==2.1.0
+py-cpuinfo==8.0.0
 psutil==5.8.0
 gunicorn==20.1.0
 black==20.8b1
 flake8==3.9.0
-pytest==6.2.2
\ No newline at end of file
+pytest==6.2.2