Refactored ID type mapping hierarchy and primary key API (bug 313204)

1 files changed, 2 insertions, 3 deletions

diff --git a/jpa_diagram_editor/plugins/org.eclipse.jpt.jpadiagrameditor.ui/src/org/eclipse/jpt/jpadiagrameditor/ui/internal/util/JpaArtifactFactory.java b/jpa_diagram_editor/plugins/org.eclipse.jpt.jpadiagrameditor.ui/src/org/eclipse/jpt/jpadiagrameditor/ui/internal/util/JpaArtifactFactory.java
index 0e692cc095..89cbe8fb4f 100644
--- a/jpa_diagram_editor/plugins/org.eclipse.jpt.jpadiagrameditor.ui/src/org/eclipse/jpt/jpadiagrameditor/ui/internal/util/JpaArtifactFactory.java
+++ b/jpa_diagram_editor/plugins/org.eclipse.jpt.jpadiagrameditor.ui/src/org/eclipse/jpt/jpadiagrameditor/ui/internal/util/JpaArtifactFactory.java
@@ -1,7 +1,7 @@
 /*******************************************************************************
  * <copyright>
  *
- * Copyright (c) 2005, 2012 SAP AG and others.
+ * Copyright (c) 2005, 2013 SAP AG and others.
  * All rights reserved. This program and the accompanying materials
  * are made available under the terms of the Eclipse Public License v1.0
  * which accompanies this distribution, and is available at
@@ -24,7 +24,6 @@ import java.util.LinkedList;
 import java.util.List;
 import java.util.ListIterator;
 import java.util.Set;
-
 import org.eclipse.core.resources.IProject;
 import org.eclipse.core.resources.IResource;
 import org.eclipse.core.resources.ResourcesPlugin;
@@ -1839,7 +1838,7 @@ public class JpaArtifactFactory {
 		JavaPersistentType idClassType = null;
 		TypeMapping mapping = getTypeMapping(jpt);
 		if(mapping instanceof Entity){
-			idClassType = ((Entity)mapping).getIdClass();
+			idClassType = ((Entity)mapping).getIdClassReference().getIdClass();
 			if(idClassType == null) {
 				String idClassFqn = ((Entity)mapping).getIdClassReference().getFullyQualifiedIdClassName();
 				PersistentType idClassPersistentType = jpt.getPersistenceUnit().getPersistentType(idClassFqn);