Forgot to add new module in previous commit