It's a long time since I had one that age but we clearly noticed when the kids had an upgrade and behaviour changed. I'd put it down to him enjoying his new capabilities and being to excited to sleep. Don't think it's a problem, just roll with it, it's not as if it's a toy you can take away!
Have fun, it doesn't get much easier, but it does get different.