I agree that CO in itself is a little underwhelming, but with the changes you're suggesting AST would probably become too powerful.
Also, if you can time it correctly to extend LA (and hopefully a nice expanded balance) it's actually quite useful.
The only small buff I could see working is the following: increase potency of already applied shields when using Noct by 20%.