diff --git a/src/main/java/fr/pasteur/ida/zellige/main/Main.java b/src/main/java/fr/pasteur/ida/zellige/main/Main.java
index 0be1b5d25c4afe8e372632f074828b6f5cbc9675..4b6f077304aa473bfb20f845d0aacb4164e39a80 100644
--- a/src/main/java/fr/pasteur/ida/zellige/main/Main.java
+++ b/src/main/java/fr/pasteur/ida/zellige/main/Main.java
@@ -57,9 +57,9 @@ public class Main
 
         // Input of the image.
         final String imagePath =
-        "doc/STK.tif";
+//        "doc/STK.tif";
 //        "C:\\Users\\ctrebeau\\Desktop\\MoucheAile\\STK\\STK_Mouche_c01_f0001_p005.tif";
-//        "C:\\Users\\ctrebeau\\Downloads\\phantom3b_combined.tif";
+        "C:\\Users\\ctrebeau\\Downloads\\phantom3b_combined.tif";
 //                "C:\\Users\\ctrebeau\\Desktop\\HighRes\\STK_170706_Vangl2-Lp-wt_E14.5_Phall_cochlea_01bHighRes_c01_f0001_p005.tif";
         System.out.println(imagePath);
         // Creation of the image : version with unsigned type. */
diff --git a/src/main/java/fr/pasteur/ida/zellige/surfaceConstruction/construction/ReferenceSurfaceExtraction.java b/src/main/java/fr/pasteur/ida/zellige/surfaceConstruction/construction/ReferenceSurfaceExtraction.java
index 3bd58de8119623e4ad034cd4a17295ae0c9cc090..4e886f31239265797069465c3239afdef2116e97 100644
--- a/src/main/java/fr/pasteur/ida/zellige/surfaceConstruction/construction/ReferenceSurfaceExtraction.java
+++ b/src/main/java/fr/pasteur/ida/zellige/surfaceConstruction/construction/ReferenceSurfaceExtraction.java
@@ -103,8 +103,8 @@ public class ReferenceSurfaceExtraction< T extends RealType< T > & NativeType< T
      */
     private  Pixels[][] rebuildPixelsArray( Surface surface )
     {
-        int width = surface.get().length;
-        int height = surface.get()[0].getLength();
+        int width = surface.getHeight();
+        int height = surface.getWidth();
         Pixels[][] tempCoordinates = new Pixels[ height ][ width ]; // Transposed array
         for ( int i = 0; i <= width - 1; i++ )
         {
diff --git a/src/main/java/fr/pasteur/ida/zellige/surfaceConstruction/element/Surface.java b/src/main/java/fr/pasteur/ida/zellige/surfaceConstruction/element/Surface.java
index 9d460f5c639350e4d0552a7f9487f018e553c614..2f34b1f4ae1ee2fe39762a5db35571e20b2f483b 100644
--- a/src/main/java/fr/pasteur/ida/zellige/surfaceConstruction/element/Surface.java
+++ b/src/main/java/fr/pasteur/ida/zellige/surfaceConstruction/element/Surface.java
@@ -306,7 +306,18 @@ public class Surface
     }
 
 
-
+    public int getWidth()
+    {
+        for ( SurfaceLine surfaceLine: surfaceLines
+               )
+        {
+            if (surfaceLine !=  null)
+            {
+                return surfaceLine.getLength();
+            }
+        }
+        return 0;
+    }
     public int getHeight()
     {
         return this.surfaceLines.length;