Add tests for transition