fix assertions in initial test

- using assert method
- fix identification of label elements