Infected Prison: Solve some known bugs.

Solve some known bugs.