Fixed bug in the context unit test.