Update spark tests