chore: Update to reflect directive changes