diff --git a/src/components/app-root/AppRootElement.ts b/src/components/app-root/AppRootElement.ts
index 017e92e..e218207 100644
--- a/src/components/app-root/AppRootElement.ts
+++ b/src/components/app-root/AppRootElement.ts
@@ -1,7 +1,7 @@
-import { controller, target } from '@github/catalyst';
+import { controller, target } from 'core/utils';
 import { BaseComponentElement } from 'common/';
 
-@controller
+@controller('app-root')
 class AppRootElement extends BaseComponentElement {
 	@target rootElement: HTMLElement;
 	constructor() {
diff --git a/src/components/wallet-header/WalletHeaderElement.ts b/src/components/wallet-header/WalletHeaderElement.ts
index ab55395..732a81c 100644
--- a/src/components/wallet-header/WalletHeaderElement.ts
+++ b/src/components/wallet-header/WalletHeaderElement.ts
@@ -3,7 +3,7 @@ import { BaseComponentElement } from 'common/';
 import { findMethod } from 'core/utils';
 import { WalletHeaderElementTemplate } from 'components/wallet-header';
 
-@controller
+@controller('wallet-header')
 class WalletHeaderElement extends BaseComponentElement {
 	@attr currentBalance: number;
 	@attr lastMonth: number;