diff --git a/FS22_BetterAILights.zip b/FS22_BetterAILights.zip
index 2800a59..7aec8fc 100644
Binary files a/FS22_BetterAILights.zip and b/FS22_BetterAILights.zip differ
diff --git a/src/aiBetterLights.lua b/src/aiBetterLights.lua
index 6099d42..ad8aece 100644
--- a/src/aiBetterLights.lua
+++ b/src/aiBetterLights.lua
@@ -4,7 +4,7 @@
 
 AIBetterLights = {};
 
-function AIBetterLights:updateAILights(isWorking)
+function AIBetterLights:updateAILights(superFunc, isWorking)
 	local spec          = self.spec_lights
 
 	local isCombine     = ( self.spec_combine ~= nil )
@@ -20,6 +20,11 @@ function AIBetterLights:updateAILights(isWorking)
 	if not g_currentMission.environment.isSunOn or g_currentMission.environment.weather:getIsRaining() then
 		local typeMask = spec.aiLightsTypesMask
 
+		-- Check to see if we are on a field.  This may be nessesary for courseplay.
+		-- if self.getIsOnField ~= nil and self:getIsOnField() then
+		-- 	typeMask = spec.aiLightsTypesMaskWork
+		-- end
+
 		if isWorking then
 			typeMask = spec.aiLightsTypesMaskWork
 		end