Updated mapping for hibernte diagram test