Fixed referencing the wrong class in automodule