Whoa missed some stuff
Forgot to fix main