Specialize test suites in reader package to classes.

1 files changed, 2 insertions, 2 deletions

diff --git a/src/cz/crcs/ectester/reader/output/YAMLTestWriter.java b/src/cz/crcs/ectester/reader/output/YAMLTestWriter.java
index e0038fd..cd89b1a 100644
--- a/src/cz/crcs/ectester/reader/output/YAMLTestWriter.java
+++ b/src/cz/crcs/ectester/reader/output/YAMLTestWriter.java
@@ -7,7 +7,7 @@ import cz.crcs.ectester.reader.command.Command;
 import cz.crcs.ectester.reader.response.Response;
 import cz.crcs.ectester.common.test.Test;
 import cz.crcs.ectester.reader.test.CommandTest;
-import cz.crcs.ectester.reader.test.TestSuite;
+import cz.crcs.ectester.reader.test.CardTestSuite;
 import org.yaml.snakeyaml.DumperOptions;
 import org.yaml.snakeyaml.Yaml;
 
@@ -31,7 +31,7 @@ public class YAMLTestWriter implements TestWriter {
     }
 
     @Override
-    public void begin(TestSuite suite) {
+    public void begin(CardTestSuite suite) {
         output.println("---");
         testRun = new HashMap<>();
         testSuite = new HashMap<>();