diff --git a/slither/tools/flattening/flattening.py b/slither/tools/flattening/flattening.py
index 90b751afb..e88c10e3a 100644
--- a/slither/tools/flattening/flattening.py
+++ b/slither/tools/flattening/flattening.py
@@ -151,12 +151,8 @@ class Flattening:
         if self._convert_library_to_internal and contract.is_library:
             for f in contract.functions_declared:
                 visibility = ""
-                if f.visibility == "external":
-                    visibility = f.visibility
-                elif f.visibility == "public":
+                if f.visibility in ["external", "public"]:
                     visibility = f.visibility
-
-                if visibility != "":
                     attributes_start = (
                         f.parameters_src().source_mapping["start"]
                         + f.parameters_src().source_mapping["length"]