Fix remaining tests in test_timeit

Author: steve-s

graalpython/com.oracle.graal.python.test/src/tests/unittest_tags_bytecode_dsl/test_timeit.txt

@@ -37,3 +37,5 @@ test.test_timeit.TestTimeit.test_timeit_function_zero_iters @ linux-x86_64
 test.test_timeit.TestTimeit.test_timeit_globals_args @ linux-x86_64
 test.test_timeit.TestTimeit.test_timeit_zero_iters @ linux-x86_64
 test.test_timeit.TestTimeit.test_timer_empty_stmt @ linux-x86_64
+test.test_timeit.TestTimeit.test_timer_invalid_setup @ linux-x86_64
+test.test_timeit.TestTimeit.test_timer_invalid_stmt @ linux-x86_64

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/compiler/bytecode_dsl/RootNodeCompiler.java

@@ -790,7 +790,9 @@ private void visitModuleBody(StmtTy[] body, Builder b) {
                         if (bodyNode instanceof StmtTy.Expr expr) {
                             // Return the value of the last statement for interop eval.
                             beginReturn(b);
+                            boolean closeTag = beginSourceSection(expr, b);
                             expr.value.accept(statementCompiler);
+                            endSourceSection(b, closeTag);
                             endReturn(b);
                         } else {
                             bodyNode.accept(statementCompiler);