diff --git a/src/main/kotlin/com/nuculabs/dev/imagetagger/ui/controls/ImageTagsEntryControl.kt b/src/main/kotlin/com/nuculabs/dev/imagetagger/ui/controls/ImageTagsEntryControl.kt
index f6bd608..8670a37 100644
--- a/src/main/kotlin/com/nuculabs/dev/imagetagger/ui/controls/ImageTagsEntryControl.kt
+++ b/src/main/kotlin/com/nuculabs/dev/imagetagger/ui/controls/ImageTagsEntryControl.kt
@@ -58,11 +58,7 @@ class ImageTagsEntryControl
      */
     fun setImage(imagePath: String) {
         File(imagePath).inputStream().use {
-            imageView.image = Image(it)
-            imageView.resize(244.0, 244.0)
-            imageView.fitHeight = 244.0
-            imageView.fitWidth = 244.0
-            imageView.isSmooth = true
+            imageView.image = Image(it, 244.0, 244.0, true, true)
             imageView.isCache = true
         }
     }
diff --git a/src/main/resources/com/nuculabs/dev/imagetagger/ui/controls/image-tags-entry.fxml b/src/main/resources/com/nuculabs/dev/imagetagger/ui/controls/image-tags-entry.fxml
index 96e20fd..56d950e 100644
--- a/src/main/resources/com/nuculabs/dev/imagetagger/ui/controls/image-tags-entry.fxml
+++ b/src/main/resources/com/nuculabs/dev/imagetagger/ui/controls/image-tags-entry.fxml
@@ -10,7 +10,9 @@
         <Insets bottom="5.0" left="5.0" right="5.0" top="5.0"/>
     </padding>
     <HBox>
-        <ImageView fx:id="imageView" />
+        <StackPane style="-fx-background-color: lightgray;" prefWidth="244" prefHeight="244">
+            <ImageView fx:id="imageView" />
+        </StackPane>
         <VBox>
             <padding>
                 <Insets bottom="5.0" left="5.0" right="5.0" top="5.0"/>