Xiaomi gateway pycrypto, pycryptodome. Problem!

Trying to install Xiaomi gateway in Home Assistant but something is wrong. I use hassbian and hass 0.54.
Have tried with and without mac address, reset xiaomin and started over from the beginning but nothing works.
Anyone who has any idea what might be wrong?

Here’s how my configuration.yaml looks like:

xiaomi_aqara:
  gateways:
   - host: 192.168.1.138
     mac:
     key: xxxxxxxxxxxxxx

Gets these error messages in the log:

2017-10-09 20:25:56 ERROR (MainThread) [homeassistant.setup] Error during setup of component xiaomi_aqara
Traceback (most recent call last):
File "/srv/homeassistant/lib/python3.5/site-packages/homeassistant/setup.py", line 194, in _async_setup_component
component.setup, hass, processed_config)
File "/usr/lib/python3.5/asyncio/futures.py", line 380, in __iter__
yield self # This tells Task to wait for completion.
File "/usr/lib/python3.5/asyncio/tasks.py", line 304, in _wakeup
future.result()
File "/usr/lib/python3.5/asyncio/futures.py", line 293, in result
raise self._exception
File "/usr/lib/python3.5/concurrent/futures/thread.py", line 55, in run
result = self.fn(*self.args, **self.kwargs)
File "/srv/homeassistant/lib/python3.5/site-packages/homeassistant/components/xiaomi_aqara.py", line 87, in setup
from PyXiaomiGateway import PyXiaomiGateway
File "/srv/homeassistant/lib/python3.5/site-packages/PyXiaomiGateway/__init__.py", line 9, in <module>
from Crypto.Cipher import AES
File "/srv/homeassistant/lib/python3.5/site-packages/Crypto/Cipher/__init__.py", line 3, in <module>
from Crypto.Cipher._mode_ecb import _create_ecb_cipher
File "/srv/homeassistant/lib/python3.5/site-packages/Crypto/Cipher/_mode_ecb.py", line 29, in <module>
from Crypto.Util._raw_api import (load_pycryptodome_raw_lib,
File "/srv/homeassistant/lib/python3.5/site-packages/Crypto/Util/_raw_api.py", line 32, in <module>
from Crypto.Util.py3compat import byte_string
ImportError: cannot import name 'byte_string'

Guess that something is wrong with my installation of pycrypto or is it pycryptodome i should use? Anyone who has any idea what might be wrong and how I do to solve this?

Remove pycryptodome if it is installed: pip uninstall pycryptodome
Then install pycrypto: pip install pycrypto