Fix import in unittest for transformations.