Fixed bug where collimator wasn't being added