diff --git a/buttonPrototypes/ActionButton.lua b/buttonPrototypes/ActionButton.lua
index b949037..7354221 100644
--- a/buttonPrototypes/ActionButton.lua
+++ b/buttonPrototypes/ActionButton.lua
@@ -224,7 +224,7 @@ function Button:RefreshAllStateActions()
 end
 
 function Button:RefreshStateAction(state)
-	local state = tonumber(state or self:GetAttribute("state"))
+	local state = tonumber(state or self:GetAttribute("state") or 0)
 	local action = self.stateactions[state]
 	self:SetAttribute("type--" .. state, "action")
 	self:SetAttribute("action--" .. state, action)