diff --git a/packages/grafana-ui/src/components/BarGauge/BarGauge.test.tsx b/packages/grafana-ui/src/components/BarGauge/BarGauge.test.tsx
index 4ba1d9d1b9b..251f2ce9d4a 100644
--- a/packages/grafana-ui/src/components/BarGauge/BarGauge.test.tsx
+++ b/packages/grafana-ui/src/components/BarGauge/BarGauge.test.tsx
@@ -69,12 +69,12 @@ describe('BarGauge', () => {
   describe('Get value color', () => {
     it('should get the threshold color if value is same as a threshold', () => {
       const props = getProps();
-      props.value = props.display(70);
+      props.value = props.display!(70);
       expect(getValueColor(props)).toEqual(orange);
     });
     it('should get the base threshold', () => {
       const props = getProps();
-      props.value = props.display(-10);
+      props.value = props.display!(-10);
       expect(getValueColor(props)).toEqual(green);
     });
   });