You also have the entity as an argument, why not use that instead?