fix tests. fixed a bug in context() introduced in #101