Fix production bugs