diff --git a/salt/switches/TL-SG3210.expect b/salt/switches/TL-SG3210.expect
index b47d547..b4586b2 100644
--- a/salt/switches/TL-SG3210.expect
+++ b/salt/switches/TL-SG3210.expect
@@ -90,6 +90,8 @@ expect "(config-if-range)#"
 {%- endfor %}
 send "switchport trunk allowed vlan {{ ','.join(vlan_ids) }}\r"
 expect "(config-if-range)#"
+send "exit\r"
+expect "(config)#"
 {%- elif conf['mode'] == 'access' %}
 {%- for port in conf['ports'] %}
 send "interface range gigabitEthernet 1/0/{{ port }}\r"
@@ -98,11 +100,11 @@ send "switchport mode access\r"
 expect "(config-if-range)#"
 send "switchport access vlan {{ pillar['vlans'][name] }}\r"
 expect "(config-if-range)#"
-{%- endfor %}
-{%- endif %}
 send "exit\r"
 expect "(config)#"
 {%- endfor %}
+{%- endif %}
+{%- endfor %}
 
 send "exit\r"
 expect "#"