Fixing mini-batch normalization