diff --git a/generator/setupgenerator.h b/generator/setupgenerator.h
index 5c7eff2..84dc0ce 100644
--- a/generator/setupgenerator.h
+++ b/generator/setupgenerator.h
@@ -60,7 +60,7 @@ class SetupGenerator : public Generator
   
  private:
    QStringList writePolymorphicHandler(QTextStream &s, const QString &package,
-     const AbstractMetaClassList &classes);
+     const AbstractMetaClassList &polyBaseClasses, QList<const AbstractMetaClass*>& allClasses);
 
    QHash<QString, QList<const AbstractMetaClass*> > packHash;
 };