diff --git a/aimodel/slurm-pretrain-plot.job b/aimodel/slurm-pretrain-plot.job
index cda2159..1d0c6e9 100755
--- a/aimodel/slurm-pretrain-plot.job
+++ b/aimodel/slurm-pretrain-plot.job
@@ -36,7 +36,7 @@ if [[ -z "${INPUT}" ]]; then
 	exit 0;
 fi
 
-if [[ ! -d "${INPUT}" ]]; then
+if [[ ! -r "${INPUT}" ]]; then
 	echo -e "Error: The input filepath at '${INPUT}' containing the input .tfrecord dataset either doesn't exist or isn't a directory.";
 	show_help;
 	exit 1;