diff --git a/test/aeso_compiler_tests.erl b/test/aeso_compiler_tests.erl
index 561254c..cb4d047 100644
--- a/test/aeso_compiler_tests.erl
+++ b/test/aeso_compiler_tests.erl
@@ -208,6 +208,7 @@ compilable_contracts() ->
      "polymorphism_contract_interface_extensions",
      "polymorphism_contract_interface_same_decl_multi_interface",
      "polymorphism_contract_interface_same_name_same_type",
+     "missing_init_fun_state_unit",
      "test" % Custom general-purpose test file. Keep it last on the list.
     ].
 
@@ -1035,9 +1036,14 @@ failing_contracts() ->
                   [<<?Pos(3,12)
                      "The event type cannot be parameterized">>
                   ])
-    , ?TYPE_ERROR(missing_init_function,
+    , ?TYPE_ERROR(missing_init_fun_alias_to_type,
                   [<<?Pos(1,10)
-                     "Missing `init` function for the contract `MissingInitFunction`.\n"
+                     "Missing `init` function for the contract `AliasToType`.\n"
+                     "The `init` function can only be omitted if the state type is `unit`">>
+                  ])
+    , ?TYPE_ERROR(missing_init_fun_alias_to_alias_to_type,
+                  [<<?Pos(1,10)
+                     "Missing `init` function for the contract `AliasToAliasToType`.\n"
                      "The `init` function can only be omitted if the state type is `unit`">>
                   ])
     ].
diff --git a/test/contracts/missing_init_fun_alias_to_alias_to_type.aes b/test/contracts/missing_init_fun_alias_to_alias_to_type.aes
new file mode 100644
index 0000000..7514fd0
--- /dev/null
+++ b/test/contracts/missing_init_fun_alias_to_alias_to_type.aes
@@ -0,0 +1,3 @@
+contract AliasToAliasToType =
+  type alias = int * int
+  type state = alias
\ No newline at end of file
diff --git a/test/contracts/missing_init_fun_alias_to_type.aes b/test/contracts/missing_init_fun_alias_to_type.aes
new file mode 100644
index 0000000..3e0b828
--- /dev/null
+++ b/test/contracts/missing_init_fun_alias_to_type.aes
@@ -0,0 +1,2 @@
+contract AliasToType =
+  type state = int * int
\ No newline at end of file
diff --git a/test/contracts/missing_init_fun_state_unit.aes b/test/contracts/missing_init_fun_state_unit.aes
new file mode 100644
index 0000000..04603e2
--- /dev/null
+++ b/test/contracts/missing_init_fun_state_unit.aes
@@ -0,0 +1,9 @@
+contract AliasToAliasToUnit =
+  type alias = unit
+  type state = alias
+
+contract AliasToUnit =
+  type state = unit
+
+main contract ImplicitState =
+    type sometype = int
\ No newline at end of file
diff --git a/test/contracts/missing_init_function.aes b/test/contracts/missing_init_function.aes
deleted file mode 100644
index 49372fb..0000000
--- a/test/contracts/missing_init_function.aes
+++ /dev/null
@@ -1,3 +0,0 @@
-contract MissingInitFunction =
-  type state = int * int
-