Correct type in test entity