diff --git a/rso.go b/rso.go
index 2bd6d23..8ebc132 100644
--- a/rso.go
+++ b/rso.go
@@ -191,7 +191,11 @@ func (r *rover) PopulateModuleState(rso *ResourcesOverview, module *tfjson.State
 				if _, ok := rs[parent]; !ok {
 					rs[parent] = &StateOverview{}
 					rs[parent].Children = make(map[string]*StateOverview)
-					rs[parent].Type = ResourceTypeResource
+					if rst.Mode == "data" {
+						rs[parent].Type = ResourceTypeData
+					} else {
+						rs[parent].Type = ResourceTypeResource
+					}
 
 				}