Add context as a keyword for tests