diff --git a/packages/diagrams-demo-gallery/demos/demo-custom-node1/DiamondNodeFactory.tsx b/packages/diagrams-demo-gallery/demos/demo-custom-node1/DiamondNodeFactory.tsx
index 8b5e2a0..675149e 100644
--- a/packages/diagrams-demo-gallery/demos/demo-custom-node1/DiamondNodeFactory.tsx
+++ b/packages/diagrams-demo-gallery/demos/demo-custom-node1/DiamondNodeFactory.tsx
@@ -1,4 +1,4 @@
-import { DiamonNodeWidget } from './DiamondNodeWidget';
+import { DiamondNodeWidget } from './DiamondNodeWidget';
import { DiamondNodeModel } from './DiamondNodeModel';
import * as React from 'react';
import { AbstractReactFactory } from '@projectstorm/react-canvas-core';
@@ -10,7 +10,7 @@ export class DiamondNodeFactory extends AbstractReactFactory;
+ return ;
}
generateModel(event) {
diff --git a/packages/diagrams-demo-gallery/demos/demo-custom-node1/DiamondNodeWidget.tsx b/packages/diagrams-demo-gallery/demos/demo-custom-node1/DiamondNodeWidget.tsx
index 095673f..f19a54d 100644
--- a/packages/diagrams-demo-gallery/demos/demo-custom-node1/DiamondNodeWidget.tsx
+++ b/packages/diagrams-demo-gallery/demos/demo-custom-node1/DiamondNodeWidget.tsx
@@ -3,7 +3,7 @@ import { DiamondNodeModel } from './DiamondNodeModel';
import { DiagramEngine, PortModelAlignment, PortWidget } from '@projectstorm/react-diagrams';
import styled from '@emotion/styled';
-export interface DiamonNodeWidgetProps {
+export interface DiamondNodeWidgetProps {
node: DiamondNodeModel;
engine: DiagramEngine;
size?: number;
@@ -27,7 +27,7 @@ namespace S {
/**
* @author Dylan Vorster
*/
-export class DiamonNodeWidget extends React.Component {
+export class DiamondNodeWidget extends React.Component {
render() {
return (