Add tests for the base element