Just follow the steps in reverse.

Fit the belt onto the white gear on the brush assembly. With white end cap pulled out, fit the brush axel into the brush assembly, make sure the belt is in the right place and runs freely. Push the white end cap closed.