Fix code highlight for setup and teardown sample