diff --git a/.idea/runConfigurations/MockGeneratorTest.xml b/.idea/runConfigurations/MockGeneratorTest.xml
index cd48c5f..9e8a816 100644
--- a/.idea/runConfigurations/MockGeneratorTest.xml
+++ b/.idea/runConfigurations/MockGeneratorTest.xml
@@ -1,9 +1,8 @@
 <component name="ProjectRunConfigurationManager">
   <configuration default="false" name="MockGeneratorTest" type="JUnit" factoryName="JUnit" singleton="true">
-    <extension name="coverage" enabled="false" merge="false" sample_coverage="true" runner="idea" />
     <module name="MockGenerator" />
     <option name="ALTERNATIVE_JRE_PATH_ENABLED" value="true" />
-    <option name="ALTERNATIVE_JRE_PATH" value="OC-173.4674" />
+    <option name="ALTERNATIVE_JRE_PATH" value="OC-181.4445" />
     <option name="PACKAGE_NAME" value="" />
     <option name="MAIN_CLASS_NAME" value="" />
     <option name="METHOD_NAME" value="" />
@@ -11,12 +10,10 @@
     <option name="VM_PARAMETERS" value="-ea" />
     <option name="PARAMETERS" value="" />
     <option name="WORKING_DIRECTORY" value="file://$MODULE_DIR$" />
-    <option name="ENV_VARIABLES" />
     <option name="PASS_PARENT_ENVS" value="true" />
     <option name="TEST_SEARCH_SCOPE">
       <value defaultName="singleModule" />
     </option>
-    <envs />
     <patterns />
     <method />
   </configuration>
diff --git a/Makefile b/Makefile
index 9f11b57..f44930f 100644
--- a/Makefile
+++ b/Makefile
@@ -1,4 +1,4 @@
-APPCODE_BUILD=173.3727
+APPCODE_BUILD=181.4445
 
 .PHONY: bootstrap downloadcommunity updateandroid updateandroidtools buildcommunity ant
 
diff --git a/MockGenerator/MockGenerator.iml b/MockGenerator/MockGenerator.iml
index ba91365..9f2d470 100644
--- a/MockGenerator/MockGenerator.iml
+++ b/MockGenerator/MockGenerator.iml
@@ -9,7 +9,7 @@
       <sourceFolder url="file://$MODULE_DIR$/tests" isTestSource="true" />
       <sourceFolder url="file://$MODULE_DIR$/testData" type="java-test-resource" />
     </content>
-    <orderEntry type="jdk" jdkName="OC-173.4674" jdkType="IDEA JDK" />
+    <orderEntry type="jdk" jdkName="OC-181.4668" jdkType="IDEA JDK" />
     <orderEntry type="sourceFolder" forTests="false" />
     <orderEntry type="module" module-name="MockGeneratorUseCases" />
     <orderEntry type="library" scope="TEST" name="junit:junit:4.12" level="project" />
diff --git a/MockGenerator/resources/META-INF/plugin.xml b/MockGenerator/resources/META-INF/plugin.xml
index 06bb143..68cb0ae 100644
--- a/MockGenerator/resources/META-INF/plugin.xml
+++ b/MockGenerator/resources/META-INF/plugin.xml
@@ -1,7 +1,7 @@
 <idea-plugin>
     <id>codes.seanhenry.mockgenerator</id>
     <name>Swift Mock Generator for AppCode</name>
-    <version>11</version>
+    <version>12</version>
     <vendor email="hello@seanhenry.codes" url="https://github.com/seanhenry/MockGenerator">Sean Henry</vendor>
 
     <description><![CDATA[
@@ -66,14 +66,13 @@ class MockDataStore: DataStore {
 
     <change-notes><![CDATA[
       <ul>
-        <li>Supports capturing of generic method parameters.</li>
-        <li>Supports capturing of generic method return values.</li>
+        <li>Fixes missing method error in AppCode 2018.1.2.</li>
       </ul>
     ]]>
     </change-notes>
 
     <!-- please see http://www.jetbrains.org/intellij/sdk/docs/basics/getting_started/build_number_ranges.html for description -->
-    <idea-version since-build="173.2463"/>
+    <idea-version since-build="181.4668"/>
 
     <actions>
     </actions>
diff --git a/MockGenerator/src/codes/seanhenry/util/finder/SwiftTypeItemFinder.java b/MockGenerator/src/codes/seanhenry/util/finder/SwiftTypeItemFinder.java
index ab52ce4..5723cdd 100644
--- a/MockGenerator/src/codes/seanhenry/util/finder/SwiftTypeItemFinder.java
+++ b/MockGenerator/src/codes/seanhenry/util/finder/SwiftTypeItemFinder.java
@@ -71,8 +71,10 @@ private List<SwiftTypeDeclaration> getResolvedTypes(SwiftTypeDeclaration typeDec
     if (inheritanceClause == null) {
       return Collections.emptyList();
     }
-    List<SwiftTypeDeclaration> results = inheritanceClause.getReferenceTypeElementList()
+    List<SwiftTypeDeclaration> results = inheritanceClause.getTypeElementList()
         .stream()
+        .filter((e) -> e instanceof SwiftReferenceTypeElement)
+        .map((e) -> (SwiftReferenceTypeElement) e)
         .map(this::getResolvedType)
         .filter(Objects::nonNull)
         .collect(Collectors.toList());
diff --git a/MockGenerator/tests/codes/seanhenry/testhelpers/ImportProjectTestCase.java b/MockGenerator/tests/codes/seanhenry/testhelpers/ImportProjectTestCase.java
index 8f082a0..322a9be 100644
--- a/MockGenerator/tests/codes/seanhenry/testhelpers/ImportProjectTestCase.java
+++ b/MockGenerator/tests/codes/seanhenry/testhelpers/ImportProjectTestCase.java
@@ -1,51 +1,52 @@
 package codes.seanhenry.testhelpers;
 
 import com.intellij.codeInsight.intention.IntentionAction;
+import com.intellij.idea.IdeaTestApplication;
 import com.intellij.openapi.command.WriteCommandAction;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.vfs.newvfs.impl.VfsRootAccess;
+import com.intellij.openapi.vfs.newvfs.persistent.PersistentFS;
+import com.intellij.openapi.vfs.newvfs.persistent.PersistentFSImpl;
 import com.intellij.psi.PsiFile;
 import com.intellij.psi.impl.source.tree.injected.InjectedLanguageUtil;
-import com.intellij.testFramework.PlatformTestCase;
+import com.intellij.testFramework.UsefulTestCase;
 import com.intellij.testFramework.fixtures.CodeInsightTestFixture;
 
-import java.io.*;
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.util.Arrays;
 import java.util.stream.Collectors;
 
-public abstract class ImportProjectTestCase extends PlatformTestCase {
+public abstract class ImportProjectTestCase extends UsefulTestCase {
 
   private Path testResultPath;
   private CodeInsightTestFixture fixture;
 
   @Override
   protected void tearDown() throws Exception {
-    try {
       fixture.tearDown();
-    } catch (Throwable e) {
-      e.printStackTrace();
-    } finally {
       fixture = null;
-      try {
-        super.tearDown();
-      } catch (Throwable e) {
-        e.printStackTrace();
-      }
-    }
+      super.tearDown();
   }
 
   @Override
   protected void setUp() throws Exception {
     allowAccessToXcodeDirectory();
     testResultPath = Files.createTempDirectory("codes.seanhenry.mockgenerator");
-    try {
-      super.setUp();
-    } catch (Throwable e) {
-      e.printStackTrace();
-    }
+    super.setUp();
+    this.initApplication();
+    fixture = ImportProjectTestFixtureFactory.getFixtureFactory().createCodeInsightFixture(getDataPath(), getProjectFileName());
+    fixture.setUp();
+  }
+
+  private void initApplication() throws Exception {
+     IdeaTestApplication.getInstance(null);
+        ((PersistentFSImpl)PersistentFS.getInstance()).cleanPersistedContents();
   }
 
   private void allowAccessToXcodeDirectory() throws IOException {
@@ -60,13 +61,6 @@ private String findXcodePath() throws IOException {
     return input.lines().collect(Collectors.joining());
   }
 
-  @Override
-  protected void setUpProject() throws Exception {
-    super.setUpProject();
-    fixture = ImportProjectTestFixtureFactory.getFixtureFactory().createCodeInsightFixture(getDataPath(), getProjectFileName());
-    fixture.setUp();
-  }
-
   protected abstract String getDataPath();
   protected abstract String getProjectFileName();
 
diff --git a/MockGenerator/tests/codes/seanhenry/testhelpers/ImportProjectTestFixture.java b/MockGenerator/tests/codes/seanhenry/testhelpers/ImportProjectTestFixture.java
index a9de5b7..c9494b2 100644
--- a/MockGenerator/tests/codes/seanhenry/testhelpers/ImportProjectTestFixture.java
+++ b/MockGenerator/tests/codes/seanhenry/testhelpers/ImportProjectTestFixture.java
@@ -1,11 +1,9 @@
 package codes.seanhenry.testhelpers;
 
 import com.intellij.ide.impl.ProjectUtil;
-import com.intellij.idea.IdeaTestApplication;
 import com.intellij.openapi.module.Module;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.project.ex.ProjectManagerEx;
-import com.intellij.testFramework.LightPlatformTestCase;
 import com.intellij.testFramework.fixtures.IdeaProjectTestFixture;
 import com.intellij.testFramework.fixtures.TempDirTestFixture;