Adding setClass method to Action, Entity and EntityLink