Got it to work.
If anyone is interested you just have to make sure the new kernel is detected int he installation file.
You can do this by assigning the correct kernel variable.
locate currently installed kernels (may be different to running kernel if
it’s just been updated)
base_ver=$(get_kernel_version)
base_ver=${base_ver%%[-+]*}
kernels=“PUT YOUR KERNEL HERE”