Fix import to make api tests green again
Merge pull request #585 from elyezer/fix-api-tests

Fix import to make api tests green again