fix shrink bug