diff --git a/LandscapeStratificationMetric.py b/LandscapeStratificationMetric.py
index fecf357d42eba6082b1be46d379de7543476a9ef..2e2bb77269926bb8e1a58fe2e3f160f525487316 100644
--- a/LandscapeStratificationMetric.py
+++ b/LandscapeStratificationMetric.py
@@ -17,7 +17,7 @@ from qgis.core import (QgsProcessingAlgorithm,
                        QgsProcessingParameterFile,
                        QgsProcessingParameterString,
                        QgsProcessingParameterNumber,
-                       QgsProcessingParameterVectorLayer,
+                       QgsProcessingParameterRasterDestination,
                        QgsProcessingContext)
 from qgis import processing
 import os
@@ -226,9 +226,9 @@ class LandscapeStratification(QgsProcessingAlgorithm):
                 defaultValue=5))
 
         self.addParameter(
-            QgsProcessingParameterFolderDestination(
+            QgsProcessingParameterRasterDestination(
                 self.OUTPUT,
-                self.tr('Output folder')
+                self.tr('Output landscape stratification metric')
             )
         )
 
@@ -246,12 +246,14 @@ class LandscapeStratification(QgsProcessingAlgorithm):
             context
         )
 
-        output_folder = self.parameterAsString(
+        output_raster = self.parameterAsString(
             parameters,
             self.OUTPUT,
             context
         )
 
+        output_folder = os.path.dirname(output_raster)
+
         prefix = parameters['PREFIX'] if parameters['PREFIX'] != None else ''
         if len(prefix)>0 and prefix[-1] != '_':
             prefix+='_'