diff --git a/basic-wallet/package.json b/basic-wallet/package.json
index e0db131..259f95b 100644
--- a/basic-wallet/package.json
+++ b/basic-wallet/package.json
@@ -19,7 +19,7 @@
   },
   "scripts": {
     "start": "react-scripts start",
-    "waterloo-start-basic": "react-scripts start",
+    "start-basic": "react-scripts start",
     "build": "react-scripts build",
     "test": "react-scripts test",
     "eject": "react-scripts eject"
diff --git a/my-plugin/src/MyPlugin.ts b/my-plugin/src/MyPlugin.ts
index 163a8ad..ef8f4a6 100644
--- a/my-plugin/src/MyPlugin.ts
+++ b/my-plugin/src/MyPlugin.ts
@@ -1,4 +1,4 @@
-import { BurnerPluginContext, Plugin } from '@burner-wallet/types';
+import { BurnerPluginContext, Plugin, Actions } from '@burner-wallet/types';
 import MyPage from './ui/MyPage';
 import MyElement from './ui/MyElement';
 
@@ -7,7 +7,7 @@ interface PluginActionContext {
 }
 
 export default class MyPlugin implements Plugin {
-  private pluginContext: BurnerPluginContext;
+  private pluginContext?: BurnerPluginContext;
 
   initializePlugin(pluginContext: BurnerPluginContext) {
     this.pluginContext = pluginContext;
diff --git a/my-plugin/src/ui/MyElement.tsx b/my-plugin/src/ui/MyElement.tsx
index da9b6d4..81d1fd4 100644
--- a/my-plugin/src/ui/MyElement.tsx
+++ b/my-plugin/src/ui/MyElement.tsx
@@ -1,8 +1,8 @@
 import React, { useEffect, useState } from 'react';
-import { BurnerContext } from '@burner-wallet/types';
+import { PluginElementContext } from '@burner-wallet/types';
 import MyPlugin from '../MyPlugin';
 
-const MyElement: React.FC<BurnerContext> = ({ plugin }) => {
+const MyElement: React.FC<PluginElementContext> = ({ plugin }) => {
   const [block, setBlock] = useState<string | null>(null);
   const _plugin = plugin as MyPlugin;