diff --git a/source/interprocedural_analyses/taint/test/modelTest.ml b/source/interprocedural_analyses/taint/test/modelTest.ml
index 58f31ad4d1..d7bb191775 100644
--- a/source/interprocedural_analyses/taint/test/modelTest.ml
+++ b/source/interprocedural_analyses/taint/test/modelTest.ml
@@ -15,13 +15,9 @@ module PyrePysaEnvironment = Analysis.PyrePysaEnvironment
 
 let get_stubs_and_definitions ~source_file_name ~project =
   let pyre_api = Test.ScratchProject.pyre_pysa_read_only_api project in
-  let type_environment = Test.ScratchProject.type_environment project in
-  let source_code_api =
-    Analysis.TypeEnvironment.ReadOnly.get_untracked_source_code_api type_environment
-  in
   let qualifier = Ast.Reference.create (String.chop_suffix_exn source_file_name ~suffix:".py") in
   let ast_source =
-    Analysis.SourceCodeApi.source_of_qualifier source_code_api qualifier
+    PyrePysaEnvironment.ReadOnly.source_of_qualifier pyre_api qualifier
     |> fun option -> Option.value_exn option
   in
   let initial_callables =