New death screen does not show new layout buttons when using add-ons/resource packs

If you die in the game with resource packs applied, the new death screen appears but with the old UI layout. 

Steps to reproduce:

1. Create/open a new world with add-ons/resource packs applied
2. Die

Observed result:

Expected result: