I have to update the context with 'module'. How can I achieve that setting a class attribute?