diff --git a/src/libdanbooru/autotests/test_danbooru_post.cpp b/src/libdanbooru/autotests/test_danbooru_post.cpp
index ab11c31..cf4a0b8 100644
--- a/src/libdanbooru/autotests/test_danbooru_post.cpp
+++ b/src/libdanbooru/autotests/test_danbooru_post.cpp
@@ -21,11 +21,14 @@
 #include "test_danbooru_post.h"
 #include "../danboorupost.h"
 #include "../utils.h"
+#include "../servicebase.h"
 
 #include <QFile>
 #include <QJsonDocument>
 #include <QJsonObject>
 
+
+
 void TestDanbooruPost::initTestCase()
 {
 
@@ -42,7 +45,14 @@ void TestDanbooruPost::testPostLoading()
 
     bool ok;
     auto data = Danbooru::parseDanbooruResult(file.readAll(), &ok).toList().at(0).toMap();
-    auto post = new Danbooru::DanbooruPost(data);
+
+    if (!ok) {
+        QFAIL("JSON file failed to load");
+    }
+
+    QFETCH(Danbooru::ApiType, apiType);
+
+    auto post = new Danbooru::DanbooruPost(data, apiType);
 
     QFETCH(int, postId);
     QFETCH(int, postWidth);
@@ -70,6 +80,7 @@ void TestDanbooruPost::testPostLoading_data()
 {
     //TODO: Add API type
     QTest::addColumn<QString>("jsonFile");
+    QTest::addColumn<Danbooru::ApiType>("apiType");
     QTest::addColumn<int>("postId");
     QTest::addColumn<int>("postWidth");
     QTest::addColumn<int>("postHeight");
@@ -85,6 +96,7 @@ void TestDanbooruPost::testPostLoading_data()
 
     // Konachan.com / yande.re
     QTest::newRow("konachan") << "konachan.json" // jsonFile
+                              << Danbooru::ApiType::KonachanDanbooru
                               << 205621     // id
                               << 1280       // width
                               << 720        // height