A couple things about this. It seems odd API-wise for `get_module_by_unique_id` to throw a `KeyError`. What does that error mean? The only place (i think) that calls this (below) catches `ModuleNotFoundException` which is the only thing I'd expect that method to throw.