diff --git a/app/configuration/controllers/virtual-media-controller.html b/app/configuration/controllers/virtual-media-controller.html
index 22c4bf8..e756986 100644
--- a/app/configuration/controllers/virtual-media-controller.html
+++ b/app/configuration/controllers/virtual-media-controller.html
@@ -13,12 +13,14 @@
           <!-- Button -->
           <label class="vm__upload-choose-label">
             <input id="file-upload" type="file" file="device.file" class="hide" ng-disabled="device.isActive"/>
-            <span class="vm__upload-choose-button button btn-secondary" ng-class="{disabled:device.isActive}">Choose file</span>
+            <span class="vm__upload-choose-button btn btn-secondary" ng-class="{disabled:device.isActive}">Choose file</span>
           </label>
           <div class="vm__upload-name">
             <span ng-if="!device.file">No file selected</span>
             <span ng-if="device.file.name !== undefined">{{device.file.name}}</span>
-            <span class="icon__exit" ng-if="device.file && !device.isActive" ng-click="resetFile($index);"></span>
+            <span class="icon__exit" ng-if="device.file && !device.isActive" ng-click="resetFile($index);">
+              <icon file="icon-close.svg"></icon>
+            </span>
           </div>
         </div>
         <div class="vm__active-text vm__active-border" ng-if="device.isActive">
@@ -29,7 +31,7 @@
         </div>
       </div>
       <div class="vm__upload-start">
-        <input type="button" ng-value="device.isActive ? 'Stop' : 'Start'" ng-class="{disabled:!device.file}" class="button btn-primary" ng-click="device.isActive? stopVM($index) : startVM($index)"/>
+        <input type="button" ng-value="device.isActive ? 'Stop' : 'Start'" ng-class="{disabled:!device.file}" class="btn btn-primary" ng-click="device.isActive? stopVM($index) : startVM($index)"/>
       </div>
     </div>
   </div>