Production ready fixes