fixed missing use statement and extending ContextInterface