diff --git a/src/mpol/gridding.py b/src/mpol/gridding.py
index 65c55223..e4265378 100644
--- a/src/mpol/gridding.py
+++ b/src/mpol/gridding.py
@@ -11,8 +11,7 @@
 
 from mpol.coordinates import GridCoords
 from mpol.exceptions import DataError, ThresholdExceededError, WrongDimensionError
-
-from .datasets import GriddedDataset
+from mpol.datasets import GriddedDataset
 
 
 def _check_data_inputs_2d(uu=None, vv=None, weight=None, data_re=None, data_im=None):
diff --git a/test/GPU_SLURM.sh b/test/GPU_SLURM.sh
deleted file mode 100644
index b86092a4..00000000
--- a/test/GPU_SLURM.sh
+++ /dev/null
@@ -1,20 +0,0 @@
-#!/bin/bash 
-
-#SBATCH --nodes=1 
-#SBATCH --ntasks=1 
-#SBATCH --gpus=1
-#SBATCH --mem=64GB 
-#SBATCH --time=1:00:00 
-#SBATCH --account=ipc5094_c_gpu
-#SBATCH --partition=sla-prio
-
-# load appropriate modules
-module purge
-module load ffmpeg/4.3.2
-module load openmpi/4.1.1
-module load anaconda3/2021.05
-
-# load the virtual environment
-# source /storage/home/ipc5094/Documents/scripts/RML_init.sh
-
-pytest
\ No newline at end of file