Leaking spark context in the test