Fixing unit test.

zalando · Dec 15, 2018 · 4d27dde · 4d27dde
1 parent a0ffe70
commit 4d27dde
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/main/java/de/zalando/ep/zalenium/proxy/AutoStartProxySet.java b/src/main/java/de/zalando/ep/zalenium/proxy/AutoStartProxySet.java
@@ -206,12 +206,11 @@ public RemoteProxy remove(RemoteProxy proxy) {
         if (proxy instanceof DockerSeleniumRemoteProxy) {
             DockerSeleniumRemoteProxy dockerSeleniumRemoteProxy = (DockerSeleniumRemoteProxy) proxy;
             // Always try to remove the proxy from the pool - this will stop the container.
-            String proxyId = dockerSeleniumRemoteProxy.getId();
             try {
-                LOGGER.debug("Stopping removed container [{}", proxyId);
-                starter.stopContainer(proxyId);
+                LOGGER.debug("Stopping removed container [{}", dockerSeleniumRemoteProxy.getId());
+                starter.stopContainer(dockerSeleniumRemoteProxy.getContainerId());
             } catch (Exception e) {
-                LOGGER.error("Failed to stop container [" + proxyId + "].", e);
+                LOGGER.error("Failed to stop container [" + dockerSeleniumRemoteProxy.getId() + "].", e);
             }
         }
         return super.remove(proxy);