Fixed minor bug in shrink