I think you should still 3-bet pre. AK is a drawing hand. Sure, it's a premium hand. But unless we catch an A or K, it's A high. Any PP is ahead of us. I'd rather flat QQ-AA here and hope CO 4-bets lighter for value, thinking our range is capped to medium PP/broadways that cannot withstand a 4-bet. He might 4-bet and try to isolate with a range like 99+, AQo+.
As played, jamming it in is okay. Folding is fine too, as I doubt you have CO's range dominated. You are probably a slight dog to vs his range, and if he's nitty/passive you are in pretty bad shape. He has AA/KK here sometimes, which sucks especially if he has the former. You're either flipping and slightly behind, have the same hand, or are crushed.