diff --git a/addons/safemode/functions/fn_setWeaponSafety.sqf b/addons/safemode/functions/fn_setWeaponSafety.sqf
index 7aaaf6cfb3d..e8f92f98e30 100644
--- a/addons/safemode/functions/fn_setWeaponSafety.sqf
+++ b/addons/safemode/functions/fn_setWeaponSafety.sqf
@@ -4,29 +4,29 @@
  *
  * Arguments:
  * 0: Unit <OBJECT>
- * 3: State <BOOL>
+ * 1: Weapon <STRING>
+ * 2: State <BOOL>
  *
  * Return Value:
  * None
  *
  * Example:
- * [ACE_player, true] call ace_safemode_fnc_safeWeapon
+ * [ACE_player, true] call ace_safemode_fnc_setWeaponSafety
  *
- * Public: yes
+ * Public: no
  */
 
 params [
     ["_unit", objNull, [objNull]], 
+    ["_weapon", objNull, [objNull]], 
     ["_state", true, [true]]
 ];
 
 private _safedWeapons = _unit getVariable [QGVAR(safedWeapons), []];
 
-private _weapon = currentWeapon _unit;
 _weapon = configName (configFile >> "CfgWeapons" >> _weapon);
 
 private _muzzle = currentMuzzle _unit;
-_muzzle = configName (configFile >> "CfgWeapons" >> _muzzle);
 
 if !(_state isEqualTo (_weapon in _safedWeapons)) then {
     [_unit, _weapon, _muzzle] call FUNC(lockSafety);