diff --git a/types/atlaskit__tree/index.d.ts b/types/atlaskit__tree/index.d.ts
index 4d6a952b76..c864ad342d 100644
--- a/types/atlaskit__tree/index.d.ts
+++ b/types/atlaskit__tree/index.d.ts
@@ -65,7 +65,7 @@ interface TreeDraggableProvided {
     // will be null if the draggable is disabled
     dragHandleProps?: DraggableProvidedDragHandleProps;
     // The following props will be removed once we move to react 16
-    innerRef: (element?: HTMLElement) => void;
+    innerRef: (element?: HTMLElement | null) => void;
 }
 
 export interface RenderItemParams {