Add tests for assign tag fix