diff --git a/hoot-core/src/main/cpp/hoot/core/criterion/BuildingPartCriterion.h b/hoot-core/src/main/cpp/hoot/core/criterion/BuildingPartCriterion.h
index 864c4de..03e8ad2 100644
--- a/hoot-core/src/main/cpp/hoot/core/criterion/BuildingPartCriterion.h
+++ b/hoot-core/src/main/cpp/hoot/core/criterion/BuildingPartCriterion.h
@@ -28,7 +28,7 @@
#define BUILDING_PART_CRITERION_H
// hoot
-#include <hoot/core/criterion/ElementCriterion.h>
+#include <hoot/core/criterion/GeometryTypeCriterion.h>
namespace hoot
{
@@ -36,7 +36,7 @@ namespace hoot
/**
* Identifies parts of buildings
*/
-class BuildingPartCriterion : public ElementCriterion
+class BuildingPartCriterion : public GeometryTypeCriterion
{
public:
@@ -49,6 +49,12 @@ public:
virtual ElementCriterionPtr clone() { return ElementCriterionPtr(new BuildingPartCriterion()); }
virtual QString getDescription() const { return "Identifies parts of buildings"; }
+
+ virtual GeometryType getGeometryType() const
+ { return GeometryType::Polygon; }
+
+ virtual QString toString() const override
+ { return QString::fromStdString(className()).remove("hoot::"); }
};
}