From e3e6a76942807d6b93660c2ac080cf3f3abeacd7 Mon Sep 17 00:00:00 2001
From: Matt <matt.lavery@outlook.com>
Date: Thu, 16 Nov 2017 17:31:10 +1000
Subject: [PATCH] Minor fix for issue #13

---
 RunWorkload.ps1 | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/RunWorkload.ps1 b/RunWorkload.ps1
index 8b32f2a..9bef49b 100644
--- a/RunWorkload.ps1
+++ b/RunWorkload.ps1
@@ -83,7 +83,14 @@ Clear-Host
 Write-Output "SQL Workload Generator"
 Write-Output "Starting..."
 
-Import-Module .\RunWorkload
+[string]$workloadModulePath = ".\RunWorkload"
+if (-not(Test-Path -Path workloadModulePath))
+{
+    Write-Output "Cannot find RunWorkload module at path '$($workloadModulePath)'";
+    $workloadModulePath = Read-Host -prompt "Please enter path to '.\RunWorkload' Module included with this toolset"; 
+}
+
+Import-Module $workloadModulePath
 
 Invoke-SqlWorkload -SQLServer $SQLServer -Database $Database -UserName $UserName -Password $Password -TSQLFile $TSQLFile -TSQLSetupFile $TSQLSetupFile -Frequency $Frequency -Duration $Duration -Verbose;