diff --git a/packages/common/vpu-common-demo.js b/packages/common/vpu-common-demo.js
index c2ad05421e066418f4c7b3e9c5e6eaca31f4fd96..a0874dbd552cd1e068ce043d69ecdd343c86dc0b 100644
--- a/packages/common/vpu-common-demo.js
+++ b/packages/common/vpu-common-demo.js
@@ -1,14 +1,13 @@
 import {i18n} from './i18n.js';
-import {css, html, unsafeCSS} from 'lit-element';
+import {css, html, LitElement} from 'lit-element';
 import * as commonUtils from './utils.js';
 import * as commonStyles from './styles.js';
 import './vpu-mini-spinner.js';
 import './vpu-spinner.js';
 import {getIconCSS} from './vpu-icon.js';
 import './vpu-button.js';
-import VPULitElement from './vpu-lit-element.js';
 
-class VpuCommonDemo extends VPULitElement {
+class VpuCommonDemo extends LitElement {
     constructor() {
         super();
         this.lang = 'de';
@@ -80,7 +79,7 @@ class VpuCommonDemo extends VPULitElement {
 
     buttonClickHandler() {
         setTimeout(() => {
-            this._("vpu-button").stop();
+            this.shadowRoot.querySelector("vpu-button").stop();
         }, 1000);
     }