diff --git a/setup.py b/setup.py
index 02223f6..c469729 100644
--- a/setup.py
+++ b/setup.py
@@ -5,17 +5,17 @@ import uuid
 from setuptools import setup, find_packages
 from pip.req import parse_requirements
 
-__author__ = 'Shota Muto <dos9954@gmail.com>'
+__author__ = 'Piotr Pieprzycki <piotr.pieprzycki@dreamlab.pl>'
 
 install_reqs = parse_requirements('requirements.txt', session=uuid.uuid1())
 reqs = [str(ir.req) for ir in install_reqs]
 
 setup(
     name="napalm-vyos",
-    version="0.1.1",
+    version="0.1.2",
     packages=find_packages(),
-    author="Shota Muto",
-    author_email="dos9954@gmail.com",
+    author="Piotr Pieprzycki, Shota Muto",
+    author_email="piotr.pieprzycki@dreamlab.pl, dos9954@gmail.com",
     description="Network Automation and Programmability Abstraction Layer with Multivendor support",
     classifiers=[
         'Topic :: Utilities',
diff --git a/test/unit/conftest.py b/test/unit/conftest.py
index 8358f78..6b5028d 100644
--- a/test/unit/conftest.py
+++ b/test/unit/conftest.py
@@ -31,6 +31,7 @@ class PatchedVyOSDriver(vyos.VyOSDriver):
     """Patched VyOS Driver."""
 
     def __init__(self, hostname, username, password, timeout=60, optional_args=None):
+        optional_args = {'port': '12206'}
         super().__init__(hostname, username, password, timeout, optional_args)
 
         self.patched_attrs = ['device']
diff --git a/test/unit/mocked_data/test_get_bgp_neighbors/normal/expected_result.json b/test/unit/mocked_data/test_get_bgp_neighbors/normal/expected_result.json
index ecc4db8..cd2a323 100644
--- a/test/unit/mocked_data/test_get_bgp_neighbors/normal/expected_result.json
+++ b/test/unit/mocked_data/test_get_bgp_neighbors/normal/expected_result.json
@@ -1 +1 @@
-{"global": {"router_id": "...", "peers": {"10.0.1.100": {"is_enabled": true, "uptime": "...", "remote_as": 65001, "description": "", "remote_id": "...", "local_as": 65002, "is_up": true, "address_family": {"ipv4": {"sent_prefixes": -1, "accepted_prefixes": "...", "received_prefixes": "..."}}}}}}
+{"global": {"router_id": "...", "peers": {"10.0.1.100": {"is_enabled": true, "uptime": "...", "remote_as": 65001, "description": "", "remote_id": "...", "local_as": 65002, "is_up": "...", "address_family": {"ipv4": {"sent_prefixes": -1, "accepted_prefixes": "...", "received_prefixes": "..."}}}}}}