Add mock for dist (fixed dependency injection)