Fixed another problem when loading a mock module