From 5c1d18924d2a6412201963d31fe6ab9f3d79db47 Mon Sep 17 00:00:00 2001
From: David Dias <daviddias.p@gmail.com>
Date: Mon, 27 Jun 2016 10:06:09 +0100
Subject: [PATCH] assign conn to the object level scope

---
 src/connection.js | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/connection.js b/src/connection.js
index 64d4786..03458d5 100644
--- a/src/connection.js
+++ b/src/connection.js
@@ -17,7 +17,7 @@ function Connection (conn) {
   let peerInfo
 
   this.getPeerInfo = (callback) => {
-    if (conn.getPeerInfo) {
+    if (conn && conn.getPeerInfo) {
       return conn.getPeerInfo(callback)
     }
 
@@ -29,20 +29,21 @@ function Connection (conn) {
   }
 
   this.setPeerInfo = (_peerInfo) => {
-    if (conn.setPeerInfo) {
+    if (conn && conn.setPeerInfo) {
       return conn.setPeerInfo(_peerInfo)
     }
     peerInfo = _peerInfo
   }
 
   this.getObservedAddrs = (callback) => {
-    if (conn.getObservedAddrs) {
+    if (conn && conn.getObservedAddrs) {
       return conn.getObservedAddrs(callback)
     }
     callback(null, [])
   }
 
-  this.setInnerConn = (conn) => {
+  this.setInnerConn = (_conn) => {
+    conn = _conn
     this.setReadable(conn)
     this.setWritable(conn)
   }