Found the issue, did a small fix