Updated for WeaponModel enum.

Signed-off-by: mk124 <mk124a@gmail.com>