diff --git a/src/Demo.vue b/src/Demo.vue
index 4844938..784445a 100644
--- a/src/Demo.vue
+++ b/src/Demo.vue
@@ -6,7 +6,7 @@
 </template>
 
 <script>
-import Mockup from '../bundle/vue-three-d-mockup.esm';
+import Mockup from './Mockup.vue';
 
 export default {
   components: {
diff --git a/vue.config.js b/vue.config.js
index 9abd69d..6d49d1f 100644
--- a/vue.config.js
+++ b/vue.config.js
@@ -1,5 +1,5 @@
 module.exports = {
-  publicPath: '/vue-three-d-mockup/',
+  publicPath: process.env.NODE_ENV === 'production' ? '/vue-three-d-mockup/' : '/',
   chainWebpack: (config) => {
     config.module
       .rule('file-loader')