Minor fixes

Author: montyly

slither/tools/flattening/flattening.py

@@ -191,7 +191,7 @@ def _pragmas(self) -> str:
             ret += f"pragma solidity {self._pragma_solidity};\n"
         else:
             # TODO support multiple compiler version
-            ret += f"pragma solidity {self._slither.crytic_compile.compilation_units[0].compiler_version.version};\n"
+            ret += f"pragma solidity {list(self._slither.crytic_compile.compilation_units.values())[0].compiler_version.version};\n"
 
         if self._use_abi_encoder_v2:
             ret += "pragma experimental ABIEncoderV2;\n"

tests/test_function.py

@@ -245,7 +245,9 @@ def test_functions():
 def test_function_can_send_eth():
     slither = Slither("tests/test_function.sol")
     compilation_unit = slither.compilation_units[0]
-    functions = compilation_unit.contracts_as_dict["TestFunctionCanSendEth"].available_functions_as_dict()
+    functions = compilation_unit.contracts_as_dict[
+        "TestFunctionCanSendEth"
+    ].available_functions_as_dict()
 
     assert functions["send_direct()"].can_send_eth() is True
     assert functions["transfer_direct()"].can_send_eth() is True