diff --git a/src/smartplaylists/smartplaylistquerywizardplugin.h b/src/smartplaylists/smartplaylistquerywizardplugin.h
index fb676003c..fb87af066 100644
--- a/src/smartplaylists/smartplaylistquerywizardplugin.h
+++ b/src/smartplaylists/smartplaylistquerywizardplugin.h
@@ -44,7 +44,7 @@ class SmartPlaylistQueryWizardPlugin : public SmartPlaylistWizardPlugin {
   explicit SmartPlaylistQueryWizardPlugin(Application *app, CollectionBackend *collection, QObject *parent);
   ~SmartPlaylistQueryWizardPlugin();
 
-  QString type() const { return "Query"; }
+  PlaylistGenerator::Type type() const { return PlaylistGenerator::Type_Query; }
   QString name() const;
   QString description() const;
   bool is_dynamic() const { return true; }
diff --git a/src/smartplaylists/smartplaylistwizardplugin.h b/src/smartplaylists/smartplaylistwizardplugin.h
index e5b233c28..396ae4287 100644
--- a/src/smartplaylists/smartplaylistwizardplugin.h
+++ b/src/smartplaylists/smartplaylistwizardplugin.h
@@ -23,7 +23,7 @@
 
 #include <QObject>
 
-#include "playlistgenerator_fwd.h"
+#include "playlistgenerator.h"
 
 class QWizard;
 
@@ -36,7 +36,7 @@ class SmartPlaylistWizardPlugin : public QObject {
  public:
   explicit SmartPlaylistWizardPlugin(Application *app, CollectionBackend *collection, QObject *parent);
 
-  virtual QString type() const = 0;
+  virtual PlaylistGenerator::Type type() const = 0;
   virtual QString name() const = 0;
   virtual QString description() const = 0;
   virtual bool is_dynamic() const { return false; }