diff --git a/pyicloud/services/drive.py b/pyicloud/services/drive.py
index a97f684..543a319 100644
--- a/pyicloud/services/drive.py
+++ b/pyicloud/services/drive.py
@@ -78,7 +78,7 @@ class DriveNode(object):
     @property
     def name(self):
         """Gets the node name."""
-        if self.type == "file":
+        if "extension" in self.data:
             return "%s.%s" % (self.data["name"], self.data["extension"])
         return self.data["name"]