diff --git a/tests/test_datatype.py b/tests/test_datatype.py
index fb1176f..93bad69 100644
--- a/tests/test_datatype.py
+++ b/tests/test_datatype.py
@@ -3,10 +3,15 @@ import unittest
 
 import pyfroniusreg.froniusreg as froniusreg
 
+class RegisterValue():
+    def __init__(self, value):
+        self.registers = value
+
 class TestDataTypes(unittest.TestCase):
 
     def test_int16(self):
-        assert froniusreg.int16.decode(froniusreg.int16.encode(1024)) == 1024
+        value = RegisterValue(froniusreg.int16.encode(1024)[0])
+        assert froniusreg.int16.decode(value) == 1024
 
 
 if __name__ == '__main__':