3 files changed, 6 insertions, 3 deletions
diff --git a/juick-notifications/build.gradle b/juick-notifications/build.gradle
index 10207a7f..b5b7bd89 100644
--- a/juick-notifications/build.gradle
+++ b/juick-notifications/build.gradle
@@ -1,9 +1,12 @@
 apply plugin: 'java'
 apply plugin: 'org.springframework.boot'
-apply plugin: 'io.spring.dependency-management'
 
 dependencies {
     compile project(':juick-common')
+    compile("org.springframework.boot:spring-boot-starter-websocket") {
+        exclude module: "spring-boot-starter-tomcat"
+        exclude module: "spring-boot-starter-web"
+    }
     compile 'com.ganyo:gcm-server:1.1.0'
     compile 'com.turo:pushy:0.13.4'
 }
diff --git a/juick-notifications/src/main/java/com/juick/components/Notifications.java b/juick-notifications/src/main/java/com/juick/components/Notifications.java
index 79afe8fa..1842593a 100644
--- a/juick-notifications/src/main/java/com/juick/components/Notifications.java
+++ b/juick-notifications/src/main/java/com/juick/components/Notifications.java
@@ -22,7 +22,7 @@ import com.google.android.gcm.server.*;
 import com.juick.ExternalToken;
 import com.juick.User;
 import com.juick.formatters.PlainTextFormatter;
-import com.juick.server.component.DisconnectedEvent;
+import com.juick.service.component.DisconnectedEvent;
 import com.juick.util.MessageUtils;
 import com.turo.pushy.apns.ApnsClient;
 import com.turo.pushy.apns.PushNotificationResponse;
diff --git a/juick-notifications/src/main/java/com/juick/components/service/JuickServerReconnectManager.java b/juick-notifications/src/main/java/com/juick/components/service/JuickServerReconnectManager.java
index 1f19eec9..b490c154 100644
--- a/juick-notifications/src/main/java/com/juick/components/service/JuickServerReconnectManager.java
+++ b/juick-notifications/src/main/java/com/juick/components/service/JuickServerReconnectManager.java
@@ -1,6 +1,6 @@
 package com.juick.components.service;
 
-import com.juick.server.component.DisconnectedEvent;
+import com.juick.service.component.DisconnectedEvent;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.context.ApplicationListener;