diff --git a/test/igniter/code/module_test.exs b/test/igniter/code/module_test.exs
index 465e19f..8d41a93 100644
--- a/test/igniter/code/module_test.exs
+++ b/test/igniter/code/module_test.exs
@@ -130,7 +130,7 @@ defmodule Igniter.Code.ModuleTest do
         """)
         |> Igniter.create_new_file("test.txt", "Foo")
 
-      assert {_igniter, [Foo]} =
+      assert {_igniter, [Foo, Test, TestTest]} =
                Igniter.Code.Module.find_all_matching_modules(igniter, fn _module, _zipper ->
                  true
                end)
diff --git a/test/igniter/extensions/phoenix_test.exs b/test/igniter/extensions/phoenix_test.exs
index 51c1c8f..d7239a6 100644
--- a/test/igniter/extensions/phoenix_test.exs
+++ b/test/igniter/extensions/phoenix_test.exs
@@ -59,14 +59,14 @@ defmodule Igniter.Extensions.PhoenixTest do
         end
         """)
         |> Igniter.create_new_file("lib/test_web/controllers/foo_json.ex", """
-        defmodule TestWeb.FooJson do
+        defmodule TestWeb.FooJSON do
 
           def render(_), do: %{foo: "bar"}
         end
         """)
 
       assert {:ok, "test_web/controllers/foo_json.ex"} =
-               Igniter.Extensions.Phoenix.proper_location(igniter, TestWeb.FooJson, [])
+               Igniter.Extensions.Phoenix.proper_location(igniter, TestWeb.FooJSON, [])
     end
   end
 end