fix for partial hot reload issue