Fixed 2 missing references