small changes. plus: infection should work