diff --git a/test/test_database.py b/test/test_database.py
index e43cdc2e2f0496211d46aa03a34ad3552ce6d487..d1c189791a0ff3478a22b5f8e0225fdcd4529b39 100644
--- a/test/test_database.py
+++ b/test/test_database.py
@@ -90,7 +90,7 @@ class DatabaseTest(unittest.TestCase):
         f.close()
 
         # Get data from db
-        stmt = "SELECT * FROM CodeBook"
+        stmt = "SELECT * FROM CodeBook ORDER BY Word"
         actual = self.database.conn.execute(stmt).fetchall()
 
         # Check same lenght
@@ -159,10 +159,10 @@ class DatabaseTest(unittest.TestCase):
         # Fill expected with only small codebook entries
         expected = []
         for entry in data:
-            if entry["type"] == 1:
+            if entry["type"] == 'Liten':
                 expected.append(entry)
         # Get small codebook from db
-        actual = self.database.get_codebook(codebook_size=CodebookSize.FULL)
+        actual = self.database.get_codebook(codebook_size=CodebookSize.SMALL)
         # Compare lenght
         self.assertEqual(len(expected), len(actual))
         # Get expected length of codes