The contribution presents the progress in the kinetic modelling of runaway electron distribution function for the purpose of DEMO wall-load calculations. The aim is to self-consistently simulate the phase of the disruption just after the global stochastization of the magnetic field. The tool used for the purpose is the DREAM code with different kinetic models: both full kinetic, superthermal and isotropic models were used in both toroidal and cylindrical geometries. Progress and future plans will be discussed.