A quick fix. This class will soon be removed, so no peer review needed.

cdk · Dec 9, 2011 · 9e97f2d · 9e97f2d
1 parent 3c1086a
commit 9e97f2d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/main/org/openscience/cdk/MoleculeSet.java b/src/main/org/openscience/cdk/MoleculeSet.java
@@ -141,12 +141,12 @@ public Iterable<IAtomContainer> molecules() {
      * @param  number  The position of the Molecule to be returned. 
      * @return         The Molecule at position <code>number</code> . 
      */
-    public IMolecule getMolecule(int number)
-    {
-        return (IMolecule)super.getAtomContainer(number);
+    public IMolecule getMolecule(int number) {
+        IAtomContainer container = super.getAtomContainer(number);
+        if (container instanceof IMolecule) return (IMolecule)container;
+        return getBuilder().newInstance(IMolecule.class, container);
     }
 
-
     /**
      * Returns the number of Molecules in this Container.
      *