Extraction test was moving test data to the gpu, fixed that because it shouldn't