I'll modify this to pass in the correct entity.