Fixed bug with immunity received after infection making it equal to a vaccine