From f245c524ab281d97956a972f6121a1f71618c251 Mon Sep 17 00:00:00 2001
From: Aayush Sabharwal <aayush.sabharwal@juliahub.com>
Date: Tue, 28 Nov 2023 14:38:52 +0530
Subject: [PATCH] refactor: add edge case for parameter_index

---
 src/systems/abstractsystem.jl | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/systems/abstractsystem.jl b/src/systems/abstractsystem.jl
index 7103d5dce1..48f899dd6c 100644
--- a/src/systems/abstractsystem.jl
+++ b/src/systems/abstractsystem.jl
@@ -233,6 +233,9 @@ function SymbolicIndexingInterface.is_parameter(sys::AbstractSystem, sym::Symbol
 end
 
 function SymbolicIndexingInterface.parameter_index(sys::AbstractSystem, sym)
+    if unwrap(sym) isa Int
+        return unwrap(sym)
+    end
     idx = findfirst(isequal(sym), parameters(sys))
     if idx === nothing && hasname(sym)
         idx = parameter_index(sys, getname(sym))