Replace use of WorkbenchContext with KieWorkbenchContext.
Refactored to use KieWorkbenchContext