tests: fixed entity mapping test