PDI-10451: Fixed bad imports
Merge pull request #332 from mattyb149/PDI-10451

PDI-10451: Fixed bad imports