partially fix the shrink problem