Wrong Classnames

I needed some minor fixes to get this plugin working with a wordpress source.

diff --git a/src/Plugin/migrate/source/MigrateXmlIterator.php b/src/Plugin/migrate/source/MigrateXmlIterator.php
index 807af4e..460629d 100644
--- a/src/Plugin/migrate/source/MigrateXmlIterator.php
+++ b/src/Plugin/migrate/source/MigrateXmlIterator.php
@@ -92,7 +92,7 @@ class MigrateXmlIterator implements \Iterator {
    *   The name of the MigrateXmlReader class.
    */
   public function getReaderClassName() {
-    return '\Drupal\migrate_plus\Plugin\migrate\source\MigrateXmlReader';
+    return '\Drupal\migrate_source_xml\Plugin\migrate\source\MigrateXmlReader';
   }

   /**
@@ -149,6 +149,13 @@ class MigrateXmlIterator implements \Iterator {
   /**
    * {@inheritdoc}
    */
+  public function count() {
+    return $this->xmlSource->computeCount();
+  }
+
+  /**
+   * {@inheritdoc}
+   */
   public function next() {
     $this->currentElement = NULL;
     $this->currentKey = NULL;
diff --git a/src/Plugin/migrate/source/XmlBase.php b/src/Plugin/migrate/source/XmlBase.php
index c75c683..87df3e6 100644
--- a/src/Plugin/migrate/source/XmlBase.php
+++ b/src/Plugin/migrate/source/XmlBase.php
@@ -75,7 +75,7 @@ abstract class XmlBase extends SourcePluginBase {
     parent::__construct($configuration, $plugin_id, $plugin_definition, $migration);

     if (empty($configuration['iterator_class'])) {
-      $iterator_class = '\Drupal\migrate_plus\Plugin\migrate\source\MigrateXmlIterator';
+      $iterator_class = '\Drupal\migrate_source_xml\Plugin\migrate\source\\MigrateXmlIterator';
     }
     else {
       $iterator_class = $configuration['iterator_class'];