diff --git a/src/Roster.jsx b/src/Roster.jsx
index 234a081..5472e53 100644
--- a/src/Roster.jsx
+++ b/src/Roster.jsx
@@ -142,9 +142,7 @@ const Unit = ({ unit, catalog }) => {
 	const rangedWeapons = weapons
 		.filter((weapon) => weapon.range !== "Melee" && weapon.range !== "-")
 		.sort((a, b) => a.selectionName.localeCompare(b.selectionName));
-	if (!modelStats.some((model) => model.name.includes("wounds remaining)"))) {
-		modelStats = modelStats.sort((a, b) => b.name.length - a.name.length); // Sort by length of name, so that for example "Primaris Intercessor Sergeant" is before "Primaris Intercessor"
-	}
+
 	return (
 		<div
 			className="avoid-page-break"
@@ -341,7 +339,8 @@ const ModelStats = ({
 	showWeapons,
 	modelList,
 }) => {
-	let { move, toughness, save, wounds, leadership, name, bs, ws } = modelStat;
+	let { move, toughness, save, wounds, leadership, name, bs, ws, attacks } =
+		modelStat;
 	if (!wounds) {
 		wounds = "/";
 	}
@@ -360,8 +359,22 @@ const ModelStats = ({
 	modelListMatches = modelListMatches.map((model) =>
 		model.replaceAll(name, "")
 	);
+
+	const degradingStuff = [];
 	const bsChange = Math.abs(parseInt(modelStats[0].bs, 10) - parseInt(bs, 10));
+	if (bsChange > 0) {
+		degradingStuff.push(`${bsChange} from the balistic skill`);
+	}
 	const wsChange = Math.abs(parseInt(modelStats[0].ws, 10) - parseInt(ws, 10));
+	if (wsChange > 0) {
+		degradingStuff.push(`${wsChange} from the weapon skill`);
+	}
+	const attacksChange = Math.abs(
+		parseInt(modelStats[0].attacks, 10) - parseInt(attacks, 10)
+	);
+	if (attacksChange > 0) {
+		degradingStuff.push(`${attacksChange} from the number of attacks`);
+	}
 
 	return (
 		<div style={{ display: "flex", gap: 16, alignItems: "center" }}>
@@ -395,22 +408,19 @@ const ModelStats = ({
 					))}
 				</div>
 			)}
-			{name.toLowerCase().includes("wounds remaining") &&
-				(bsChange !== 0 || wsChange !== 0) && (
-					<div
-						style={{
-							whiteSpace: "normal",
-							marginLeft: -10,
-							fontSize: "0.7em",
-						}}
-					>
-						{` Each time this model makes
+			{degradingStuff.length > 0 && (
+				<div
+					style={{
+						whiteSpace: "normal",
+						marginLeft: -10,
+						fontSize: "0.7em",
+					}}
+				>
+					{`Each time this model makes
 							an attack, subtract 
-							${bsChange ? `${bsChange} from the balistic skill` : ""}
-							${bsChange !== 0 && wsChange !== 0 ? " and " : ""}
-							${wsChange ? `${wsChange} from the weapon skill` : ""}.`}
-					</div>
-				)}
+							${degradingStuff.join(" and ")}.`}
+				</div>
+			)}
 		</div>
 	);
 };
diff --git a/src/Weapons.jsx b/src/Weapons.jsx
index c5bb7cb..86b28dc 100644
--- a/src/Weapons.jsx
+++ b/src/Weapons.jsx
@@ -228,7 +228,7 @@ const Weapon = ({ weapon, modelStats, isMelee, className }) => {
 						}}
 					>
 						{differentProfiles && selectionName + " - "}
-						{name}
+						{differentProfiles ? name.replaceAll(selectionName, "") : name}
 						{type && (
 							<span
 								style={{
diff --git a/src/assets/icons.jsx b/src/assets/icons.jsx
index 5c614f8..3bafd70 100644
--- a/src/assets/icons.jsx
+++ b/src/assets/icons.jsx
@@ -1,5 +1,5 @@
 export const Arrow = (
-	<svg viewBox="0 0 16 8" width={16} height={8} fill="#536766">
+	<svg viewBox="0 0 16 8" width={16} height={8} fill="var(--primary-color)">
 		<path d="m0 0h10l6 4-6 4h-10z" />
 	</svg>
 );
diff --git a/src/roster40k.js b/src/roster40k.js
index 066d9f2..5552192 100644
--- a/src/roster40k.js
+++ b/src/roster40k.js
@@ -277,7 +277,14 @@ export class Unit extends BaseNotes {
 	normalize() {
 		// Sort force units by role and name
 		this.models.sort(CompareModel);
-		this.modelStats.sort(CompareObj);
+		this.modelStats.sort((a, b) => {
+			if (!a.wounds) return 1;
+			var wounds_order = parseInt(a.wounds) - parseInt(b.wounds);
+			var leadership_order = parseInt(a.leadership) - parseInt(b.leadership);
+			return (
+				-wounds_order || -leadership_order || b.name.length - a.name.length
+			);
+		});
 
 		for (let model of this.models) {
 			model.normalize();