Working on quantization bug