Oh. I didn't realize that was a bug, I thought it was actually pretty neat that you could continue even after he died, if you managed to make things fall into place (e.g. on the last level I had a bomb going towards the bad guys when he died, so I was able to carry on in his honour). I would leave that in and even celebrate it ("Next level ... in his honour" or something).