Fixing small error with retention count