Use entity wrapper