diff --git a/segment-virtualenv.go b/segment-virtualenv.go
index 4df18969..7e3115a8 100644
--- a/segment-virtualenv.go
+++ b/segment-virtualenv.go
@@ -16,12 +16,15 @@ func segmentVirtualEnv(p *powerline) []pwl.Segment {
 		env = env[1 : len(env)-2]
 	}
 	if env == "" {
-		env, _ = os.LookupEnv("VIRTUAL_ENV")
-		if env != "" {
-			cfg, err := ini.Load(path.Join(env, "pyvenv.cfg"))
+		venv, _ := os.LookupEnv("VIRTUAL_ENV")
+		if venv != "" {
+			cfg, err := ini.Load(path.Join(venv, "pyvenv.cfg"))
 			if err == nil {
 				env = cfg.Section("").Key("prompt").String()
 			}
+			if env == "" {
+				env = venv
+			}
 		}
 	}
 	if env == "" {