diff --git a/src/components/icons/LayerIcon.jsx b/src/components/icons/LayerIcon.jsx
index a69c616..9f29080 100644
--- a/src/components/icons/LayerIcon.jsx
+++ b/src/components/icons/LayerIcon.jsx
@@ -19,6 +19,7 @@ class LayerIcon extends React.Component {
case 'fill-extrusion': return
case 'raster': return
case 'hillshade': return
+ case 'heatmap': return
case 'fill': return
case 'background': return
case 'line': return
diff --git a/src/components/layers/LayerEditor.jsx b/src/components/layers/LayerEditor.jsx
index 00d23d5..4764986 100644
--- a/src/components/layers/LayerEditor.jsx
+++ b/src/components/layers/LayerEditor.jsx
@@ -138,7 +138,7 @@ export default class LayerEditor extends React.Component {
onChange={v => this.changeProperty(null, 'source', v)}
/>
}
- {this.props.layer.type !== 'raster' && this.props.layer.type !== 'background' && this.props.layer.type !== 'hillshade' &&
+ {['background', 'raster', 'hillshade', 'heatmap'].indexOf(this.state.type) < 0 &&
this.setState({ source: v })}
/>
}
- {this.state.type !== 'background' && this.state.type !== 'raster' && this.state.type !== 'hillshade' &&
+ {['background', 'raster', 'hillshade', 'heatmap'].indexOf(this.state.type) < 0 &&