diff --git a/src/main/resources/META-INF/resources/frontend/code-viewer.ts b/src/main/resources/META-INF/resources/frontend/code-viewer.ts
index 9027bc3..e56a576 100644
--- a/src/main/resources/META-INF/resources/frontend/code-viewer.ts
+++ b/src/main/resources/META-INF/resources/frontend/code-viewer.ts
@@ -18,11 +18,11 @@
  * #L%
  */
 import {
-  customElement,
   html,
   LitElement
 } from "lit-element";
 
+import {customElement} from 'lit/decorators/custom-element.js';
 import "./prism.js";
 
 (function () {