Making it compatible with LArSoft env.
Minor change to make it compatible with LArSoft