Check that main is a Signal or (Signal Element), add tests for this case