1 files changed, 2 insertions, 1 deletions
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<>();