diff --git a/vital-signs/src/main/java/com/rax/vital/timer/VitalSignTimer.java b/vital-signs/src/main/java/com/rax/vital/timer/VitalSignTimer.java
index 15fa2d7..94b36ce 100644
--- a/vital-signs/src/main/java/com/rax/vital/timer/VitalSignTimer.java
+++ b/vital-signs/src/main/java/com/rax/vital/timer/VitalSignTimer.java
@@ -1,6 +1,5 @@
 package com.rax.vital.timer;
 
-import cn.hutool.core.collection.ConcurrentHashSet;
 import com.alibaba.fastjson.JSONObject;
 import com.rax.common.security.util.SecurityUtils;
 import com.rax.vital.datasource.MongoDBSource;
@@ -21,7 +20,6 @@ import java.io.IOException;
 import java.sql.Connection;
 import java.util.*;
 import java.util.concurrent.ConcurrentHashMap;
-import java.util.concurrent.ConcurrentMap;
 
 /**
  * 生命体征和用药信息推送
@@ -316,7 +314,7 @@ public class VitalSignTimer {
 	public void removeWSAIFlagSession(WebSocketSession session) {
 		userSessionMap.remove(session.getId());
 		if (userDatabaseSessionMap.containsValue(session.getId())) {
-			for (String database : userDatabaseSessionMap.values()) {
+			for (String database : userDatabaseSessionMap.keySet()) {
 				if (userDatabaseSessionMap.get(database).equals(session.getId())) {
 					userDatabaseSessionMap.remove(database);
 				}
@@ -365,10 +363,10 @@ public class VitalSignTimer {
 		machineSessionMap.put(session.getId(), session);
 	}
 
-	public void removeMachineSessionMap(WebSocketSession session) throws IOException {
+	public void removeMachineSessionMap(WebSocketSession session) throws Exception {
 		machineSessionMap.remove(session.getId());
 		if (machineDatabaseSessionMap.containsValue(session.getId())) {
-			for (String database : machineDatabaseSessionMap.values()) {
+			for (String database : machineDatabaseSessionMap.keySet()) {
 				if (machineDatabaseSessionMap.get(database).equals(session.getId())) {
 					machineDatabaseSessionMap.remove(database);
 				}