diff --git a/package.json b/package.json
index 1f53976..8ba5508 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "telegram.link",
-  "version": "0.6.2",
+  "version": "0.6.3",
   "description": "Telegram API library",
   "keywords": [
     "telegram",
diff --git a/test/telegram.link.test.js b/test/telegram.link.test.js
index 65757f3..1070bdc 100644
--- a/test/telegram.link.test.js
+++ b/test/telegram.link.test.js
@@ -185,4 +185,17 @@ describe('TelegramLink', function () {
         });
     });
 
+    describe('#un/registerOnUpdates()', function () {
+
+        it('should returns ok', function (done) {
+            var client = telegramLink.createClient({authKey: {}}, primaryDC, function () {
+                var callback = function() {};
+                client.registerOnUpdates(callback);
+                (client._channel.getParser().listeners('api.type.Updates')[0] === callback).should.be.true();
+                client.unregisterOnUpdates(callback);
+                (client._channel.getParser().listeners('api.type.Updates').length).should.be.equal(0);
+                done();
+            });
+        });
+    });
 });
diff --git a/test/utility.test.js b/test/utility.test.js
index f2b7664..5a3206b 100644
--- a/test/utility.test.js
+++ b/test/utility.test.js
@@ -5,14 +5,17 @@ var utility = require('lib/utility');
 
 describe('utility', function () {
     describe('#callService()', function () {
+        var theCallback;
         var emitter = {
             isReady: function() { return true; },
             once: function(event, callback) {
                 console.log('once ', event);
-                callback.should.be.equal('callback');
+                (typeof callback).should.be.equal('function');
+                theCallback = callback;
             },
-            emit: function(event, error) {
-                console.log('emit ', event, error.stack);
+            emit: function(event, obj) {
+                console.log('emit ', event, obj);
+                theCallback(obj);
             }
         };
         it('should returns ok', function (done) {
@@ -21,12 +24,13 @@ describe('utility', function () {
                 input.props.should.have.property('second', 2);
                 input.channel.should.be.equal('channel');
                 (typeof input.callback).should.be.equal('function');
-                done();
+                input.callback();
             };
             (function testCall(first, second, callback) {
                 var promise = utility.callService(api.service.auth.testCall, emitter, 'channel', callback, arguments);
                 promise.should.be.an.instanceOf(Promise);
-            })(1, 2, 'callback');
+                promise.then(done);
+            })(1, 2, done);
         });
     });
 });