Fixed some minor problems with infectious effect