diff --git a/modules/heartbeat_nodejs/heartbeat_nodejs.module b/modules/heartbeat_nodejs/heartbeat_nodejs.module
index 7a16cce..3f8358b 100644
--- a/modules/heartbeat_nodejs/heartbeat_nodejs.module
+++ b/modules/heartbeat_nodejs/heartbeat_nodejs.module
@@ -13,6 +13,15 @@ function heartbeat_nodejs_heartbeat_stream_load($heartbeatStream) {
 }
 
 /**
+ * Implements hook_nodejs_handlers_info().
+ */
+function heartbeat_nodejs_handlers_info() {
+  return array(
+    drupal_get_path('module', 'heartbeat_nodejs') . '/heartbeat_nodejs.js',
+  );
+}
+
+/**
  * Implements hook_heartbeat_activity_save().
  *
  * Problems on message output:
@@ -41,20 +50,6 @@ function heartbeat_nodejs_heartbeat_activity_save($heartbeatActivity) {
 
   // Create the activity message.
   $heartbeatActivity->rebuild_message();
-
-  //$accounts = nodejs_get_server_stats()->authenticatedClients;
   $output = heartbeat_activity_view($heartbeatActivity);
-
-  // Node.js push implementation.
-  nodejs_broadcast_message('', drupal_render($output));
-}
-
-
-/**
- * Implements hook_heartbeat_activity_delete().
- */
-function heartbeat_nodejs_heartbeat_activity_delete($uaids, $all) {
-  foreach ($uaids as $uaid) {
-    nodejs_broadcast_message('heartbeat-delete-message', $uaid);
-  }
+  nodejs_send_user_message($heartbeatActivity->uid_target, '', drupal_render($output));
 }