Small change to the mod class