diff --git a/spec/setup_acceptance_node.pp b/spec/setup_acceptance_node.pp
new file mode 100644
index 0000000..9a67d7d
--- /dev/null
+++ b/spec/setup_acceptance_node.pp
@@ -0,0 +1,6 @@
+# Needed for os.distro.codebase fact on ubuntu 18 on puppet 6
+if $facts['os']['name'] == 'Ubuntu' and $facts['os']['release']['full'] == '18.04' and versioncmp($facts['puppetversion'], '7') <= 0 {
+  package{'lsb-release':
+    ensure => present,
+  }
+}