Improved kernel base address handling