diff --git a/src/libdanbooru/tests/test_service.h b/src/libdanbooru/tests/test_service.h
index 53a3e4e..60dd692 100644
--- a/src/libdanbooru/tests/test_service.h
+++ b/src/libdanbooru/tests/test_service.h
@@ -17,41 +17,23 @@
  * along with Danbooru Client. If not, see <http://www.gnu.org/licenses/>.
  */
 
-#ifndef DANBOORU_TEST_CONNECT_H_
-#define DANBOORU_TEST_CONNECT_H_
+#ifndef DANBOORU_TEST_SERVICE_H_
+#define DANBOORU_TEST_SERVICE_H_
 
-#include "../konachan.h"
+#include "../moebooru.h"
+#include "../danbooru_donmai.h"
 #include "../servicebase.h"
 #include "../danboorupost.h"
 #include "../danboorupool.h"
 
 #include <QDebug>
-/*
- * Copyright 2015 Luca Beltrame <lbeltrame@kde.org>
- *
- * This file is part of Danbooru Client.
- *
- * Danbooru Client is free software: you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation, either version 3 of the License, or
- * (at your option) any later version.
- *
- * Danbooru Client is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with Danbooru Client. If not, see <http://www.gnu.org/licenses/>.
- */
-
 #include <QObject>
 
 class QLabel;
 
 namespace Danbooru {
 
-class KonachanServiceTest: public QObject
+class DanbooruServiceTest: public QObject
 {
 Q_OBJECT
 
@@ -59,8 +41,8 @@ private:
     DanbooruServiceBase* m_service;
     QLabel* m_label;
 public:
-    KonachanServiceTest(QUrl boardUrl);
-    ~KonachanServiceTest();
+    DanbooruServiceTest(QUrl boardUrl, Danbooru::ApiType type = Danbooru::MoeBooru);
+    ~DanbooruServiceTest();
 
     void testPostService();
     void testPoolService();
@@ -78,4 +60,4 @@ private Q_SLOTS:
 
 
 
-#endif
\ No newline at end of file
+#endif