fixes #23 - improper use of context in tests

@twobraids r?