Actually fixed the cobra model issue