Add context.close() on unit test