1 files changed, 7 insertions, 5 deletions
diff --git a/server/src/queryhandlerpentominos.h b/server/src/queryhandlerpentominos.h
index 9c56d07..53b01ad 100644
--- a/server/src/queryhandlerpentominos.h
+++ b/server/src/queryhandlerpentominos.h
@@ -29,26 +29,28 @@
 
 #include "queryhandler.h"
 
-#include "tcpsocket.h"
 #include "template.h"
 #include "queryresult.h"
 
-#include <vector>
 #include <string>
 
+#include <libartefact.h>
+
 /**
  * This class handles the query of external data.
  */
 class QueryHandlerPentominos : public QueryHandler {
 public:
-  QueryHandlerPentominos(TCPSocket &socket, std::string cpr);
-  ~QueryHandlerPentominos() {}
+  QueryHandlerPentominos(std::string cpr);
+  ~QueryHandlerPentominos();
 
   // Execute all queries.
   QueryResult exec(Query &query);
 
 private:
-  TCPSocket &socket;
+  atf_handle_t *atfh;
+  atf_connection_t *atfc;
+
   std::string cpr;
 };