Fixed another defmodule issue