Add mock for dist