1 files changed, 5 insertions, 19 deletions

diff --git a/examples/enterprise_test/src/enterprise_test.cpp b/examples/enterprise_test/src/enterprise_test.cpp
index 2a6b42c0..4297fc9f 100644
--- a/examples/enterprise_test/src/enterprise_test.cpp
+++ b/examples/enterprise_test/src/enterprise_test.cpp
@@ -43,30 +43,13 @@ int main(int argc, char* argv[])
 	SetBundledPluginDirectory(GetBundledPluginDirectory());
 	InitPlugins();
 
-	Ref<BinaryData> bd = BinaryData::CreateFromFilename(new FileMetadata(), argv[1]);
-	if (!bd)
-	{
-		fprintf(stderr, "Could not open input file.\n");
-		return -1;
-	}
-	Ref<BinaryView> bv;
-	for (auto type : BinaryViewType::GetViewTypes())
-	{
-		if (type->IsTypeValidForData(bd) && type->GetName() != "Raw")
-		{
-			bv = type->Create(bd);
-			break;
-		}
-	}
-
+	Ref<BinaryView> bv = BinaryNinja::Load(fname);
 	if (!bv || bv->GetTypeName() == "Raw")
 	{
-		fprintf(stderr, "Input file does not appear to be an exectuable\n");
+		fprintf(stderr, "Input file does not appear to be an executable\n");
 		return -1;
 	}
 
-	bv->UpdateAnalysisAndWait();
-
 	cout << "Target:   " << fname << endl << endl;
 	cout << "TYPE:     " << bv->GetTypeName() << endl;
 	cout << "START:    0x" << hex << bv->GetStart() << endl;
@@ -99,6 +82,9 @@ int main(int argc, char* argv[])
 			break;
 	}
 
+	// Close the file so that the resources can be freed
+	bv->GetFile()->Close();
+
 	// Shutting down is required to allow for clean exit of the core
 	BNShutdown();