diff --git a/bin/run-ci-ldc93s1_singleshotinference.sh b/bin/run-ci-ldc93s1_singleshotinference.sh
index 699b09cb..3589166b 100755
--- a/bin/run-ci-ldc93s1_singleshotinference.sh
+++ b/bin/run-ci-ldc93s1_singleshotinference.sh
@@ -14,7 +14,8 @@ fi;
 # and when trying to run on multiple devices (like GPUs), this will break
 export CUDA_VISIBLE_DEVICES=0
 
-python -u train.py --alphabet_config_path "data/alphabet.txt" \
+python -m coqui_stt_training.train \
+  --alphabet_config_path "data/alphabet.txt" \
   --show_progressbar false --early_stop false \
   --train_files ${ldc93s1_csv} --train_batch_size 1 \
   --dev_files ${ldc93s1_csv} --dev_batch_size 1 \
@@ -24,8 +25,15 @@ python -u train.py --alphabet_config_path "data/alphabet.txt" \
   --learning_rate 0.001 --dropout_rate 0.05 \
   --scorer_path 'data/smoke_test/pruned_lm.scorer'
 
-python -u train.py --alphabet_config_path "data/alphabet.txt" \
+python -m coqui_stt_training.training_graph_inference \
   --n_hidden 100 \
   --checkpoint_dir '/tmp/ckpt' \
   --scorer_path 'data/smoke_test/pruned_lm.scorer' \
   --one_shot_infer 'data/smoke_test/LDC93S1.wav'
+
+python -m coqui_stt_training.training_graph_inference_flashlight \
+  --n_hidden 100 \
+  --checkpoint_dir '/tmp/ckpt' \
+  --scorer_path 'data/smoke_test/pruned_lm.scorer' \
+  --vocab_file 'data/smoke_test/vocab.pruned.txt' \
+  --one_shot_infer 'data/smoke_test/LDC93S1.wav'
diff --git a/ci_scripts/train-extra-tests.sh b/ci_scripts/train-extra-tests.sh
index 50265afc..f538110d 100755
--- a/ci_scripts/train-extra-tests.sh
+++ b/ci_scripts/train-extra-tests.sh
@@ -69,3 +69,6 @@ time ./bin/run-ci-ldc93s1_checkpoint_bytes.sh
 
 # Training with args set via initialize_globals_from_args()
 time python ./bin/run-ldc93s1.py
+
+# Training graph inference
+time ./bin/run-ci-ldc93s1_singleshotinference.sh