diff --git a/demo-shell-ng2/app/components/files/files.component.html b/demo-shell-ng2/app/components/files/files.component.html
index 4e8906f068..e8aa437a82 100644
--- a/demo-shell-ng2/app/components/files/files.component.html
+++ b/demo-shell-ng2/app/components/files/files.component.html
@@ -8,8 +8,8 @@
         [versioning]="versioning"
         [enabled]="documentList.hasCreatePermission()">
         <div *ngIf="errorMessage" class="error-message">
-            <button (click)="resetError()" class="mdl-button mdl-js-button mdl-button--icon">
-                <i class="material-icons">highlight_off</i>
+            <button (click)="resetError()" md-icon-button>
+                <md-icon>highlight_off</md-icon>
             </button>
             <span class="error-message--text">{{errorMessage}}</span>
         </div>