3 files changed, 6 insertions, 3 deletions

diff --git a/src/cz/crcs/ectester/reader/output/TextTestWriter.java b/src/cz/crcs/ectester/reader/output/TextTestWriter.java
index 06b6ea8..b6cd51c 100644
--- a/src/cz/crcs/ectester/reader/output/TextTestWriter.java
+++ b/src/cz/crcs/ectester/reader/output/TextTestWriter.java
@@ -3,6 +3,7 @@ package cz.crcs.ectester.reader.output;
 import cz.crcs.ectester.common.output.TestWriter;
 import cz.crcs.ectester.common.test.CompoundTest;
 import cz.crcs.ectester.common.test.Test;
+import cz.crcs.ectester.common.test.TestSuite;
 import cz.crcs.ectester.reader.test.CommandTest;
 import cz.crcs.ectester.reader.test.CardTestSuite;
 
@@ -23,7 +24,7 @@ public class TextTestWriter implements TestWriter {
     }
 
     @Override
-    public void begin(CardTestSuite suite) {
+    public void begin(TestSuite suite) {
         output.println("=== Running test suite: " + suite.getName() + " ===");
         output.println("=== " + suite.getDescription());
     }
diff --git a/src/cz/crcs/ectester/reader/output/XMLTestWriter.java b/src/cz/crcs/ectester/reader/output/XMLTestWriter.java
index d889c3a..19c14f1 100644
--- a/src/cz/crcs/ectester/reader/output/XMLTestWriter.java
+++ b/src/cz/crcs/ectester/reader/output/XMLTestWriter.java
@@ -2,6 +2,7 @@ package cz.crcs.ectester.reader.output;
 
 import cz.crcs.ectester.common.output.TestWriter;
 import cz.crcs.ectester.common.test.CompoundTest;
+import cz.crcs.ectester.common.test.TestSuite;
 import cz.crcs.ectester.common.util.ByteUtil;
 import cz.crcs.ectester.reader.command.Command;
 import cz.crcs.ectester.reader.response.Response;
@@ -38,7 +39,7 @@ public class XMLTestWriter implements TestWriter {
     }
 
     @Override
-    public void begin(CardTestSuite suite) {
+    public void begin(TestSuite suite) {
         doc = db.newDocument();
         Element rootElem = doc.createElement("testSuite");
         rootElem.setAttribute("name", suite.getName());
diff --git a/src/cz/crcs/ectester/reader/output/YAMLTestWriter.java b/src/cz/crcs/ectester/reader/output/YAMLTestWriter.java
index cd89b1a..5107970 100644
--- a/src/cz/crcs/ectester/reader/output/YAMLTestWriter.java
+++ b/src/cz/crcs/ectester/reader/output/YAMLTestWriter.java
@@ -2,6 +2,7 @@ package cz.crcs.ectester.reader.output;
 
 import cz.crcs.ectester.common.output.TestWriter;
 import cz.crcs.ectester.common.test.CompoundTest;
+import cz.crcs.ectester.common.test.TestSuite;
 import cz.crcs.ectester.common.util.ByteUtil;
 import cz.crcs.ectester.reader.command.Command;
 import cz.crcs.ectester.reader.response.Response;
@@ -31,7 +32,7 @@ public class YAMLTestWriter implements TestWriter {
     }
 
     @Override
-    public void begin(CardTestSuite suite) {
+    public void begin(TestSuite suite) {
         output.println("---");
         testRun = new HashMap<>();
         testSuite = new HashMap<>();