Restore end level fireworks

Fixes a couple of issues that were preventing the fireworks from appearing at the end of the level