diff --git a/app/src/test/java/de/kuschku/quasseldroid/util/AvatarHelperTest.kt b/app/src/test/java/de/kuschku/quasseldroid/util/AvatarHelperTest.kt
index 8c9984ad47d292b6ed9962be219a064f34261ae4..c3a5c258b79d032ad7b93d2e63c694b465e70a1d 100644
--- a/app/src/test/java/de/kuschku/quasseldroid/util/AvatarHelperTest.kt
+++ b/app/src/test/java/de/kuschku/quasseldroid/util/AvatarHelperTest.kt
@@ -35,8 +35,8 @@ class AvatarHelperTest {
     val message = QuasselDatabase.MessageData(
       messageId = 1,
       time = Instant.now(),
-      type = Message_Type.of(Message_Type.Plain).toInt(),
-      flag = Message_Flag.of().toInt(),
+      type = Message_Type.of(Message_Type.Plain),
+      flag = Message_Flag.of(),
       bufferId = 0,
       sender = "justJanne",
       senderPrefixes = "",
@@ -72,8 +72,8 @@ class AvatarHelperTest {
     val message = QuasselDatabase.MessageData(
       messageId = 1,
       time = Instant.now(),
-      type = Message_Type.of(Message_Type.Plain).toInt(),
-      flag = Message_Flag.of().toInt(),
+      type = Message_Type.of(Message_Type.Plain),
+      flag = Message_Flag.of(),
       bufferId = 0,
       sender = "jwheare!sid2@irccloud.com",
       senderPrefixes = "",
@@ -109,8 +109,8 @@ class AvatarHelperTest {
     val message = QuasselDatabase.MessageData(
       messageId = 1,
       time = Instant.now(),
-      type = Message_Type.of(Message_Type.Plain).toInt(),
-      flag = Message_Flag.of().toInt(),
+      type = Message_Type.of(Message_Type.Plain),
+      flag = Message_Flag.of(),
       bufferId = 0,
       sender = "jwheare!sid2@irccloud.com",
       senderPrefixes = "",