Add tests for version handling