diff --git a/modules/device_creator.py b/modules/device_creator.py
index a389a890ab9aec21d5495f79a5872a3aae3d8518..d91bf530fa6a2535965a0af78ca2aa5602373a94 100644
--- a/modules/device_creator.py
+++ b/modules/device_creator.py
@@ -22,9 +22,9 @@ def open_file(file_name):
 def _add_provisioning(hostname, host_definitions):
     """ Adds provisioning to the device if the file exists. """
     try:
-        provision_file = open("provision/" + str(hostname) + ".yaml")
+        provision_file = open("provision/" + str(hostname) + ".yml")
         host_definitions[hostname].append("device.vm.provision \"ansible\" do |ansible|")
-        host_definitions[hostname].append("  ansible.playbook = \"provision/" + hostname + ".yaml\"")
+        host_definitions[hostname].append("  ansible.playbook = \"provision/" + hostname + ".yml\"")
         host_definitions[hostname].append("end")
     except IOError:
         pass
diff --git a/modules/provider.py b/modules/provider.py
index a8f180b1d86ef6aff016a47639096bfcfa4cff6f..c4b6941783fb31150aedb535733a408ef37628d9 100644
--- a/modules/provider.py
+++ b/modules/provider.py
@@ -54,6 +54,8 @@ def add_router_specification(router, definitions):
     router_box = "generic/debian10"
     router_memory = 256
 
+    definitions[router['name']].append(
+        "device.vm.hostname = \"" + router['name'] + "\"")
     definitions[router['name']].append(
         "device.vm.box = \"" + router_box + "\"")
     definitions[router['name']].append(