diff --git a/compiler/optimizing/codegen_test.cc b/compiler/optimizing/codegen_test.cc
index 8bb12de..af4cf73 100644
--- a/compiler/optimizing/codegen_test.cc
+++ b/compiler/optimizing/codegen_test.cc
@@ -135,8 +135,8 @@
   const DexFile::CodeItem* item = reinterpret_cast<const DexFile::CodeItem*>(data);
   HGraph* graph = builder.BuildGraph(*item);
   // Remove suspend checks, they cannot be executed in this context.
-  RemoveSuspendChecks(graph);
   ASSERT_NE(graph, nullptr);
+  RemoveSuspendChecks(graph);
   RunCodeBaseline(graph, has_result, expected);
 }