Forgot to rename module in tests