CDK-451: Fix missing import

Adding import missed during rebase on master.