Fixes flocking and a problem with awareness after the optimization pass.