From 441c6f449fe38096b43b982b1244de6821015a9b Mon Sep 17 00:00:00 2001
From: Rob McMullen <feedback@playermissile.com>
Date: Wed, 21 Sep 2016 15:24:50 -0700
Subject: [PATCH] Fix for stale code in 130k disk image support

---
 atrcopy/ataridos.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/atrcopy/ataridos.py b/atrcopy/ataridos.py
index fcf9734..288fc7f 100644
--- a/atrcopy/ataridos.py
+++ b/atrcopy/ataridos.py
@@ -249,7 +249,8 @@ class AtariDosDiskImage(DiskImageBase):
         if self.header.image_size == 133120:
             # enhanced density has 2nd VTOC
             self.vtoc2 = 1024
-            extra_free = self.get_sectors(self.vtoc2)[122:124].view(dtype='<u2')[0]
+            data, style = self.get_sectors(self.vtoc2)
+            extra_free = data[122:124].view(dtype='<u2')[0]
             self.unused_sectors += extra_free
     
     def get_directory(self):
@@ -314,7 +315,7 @@ class AtariDosDiskImage(DiskImageBase):
         segments.append(segment)
         if self.vtoc2 > 0:
             start, count = self.get_contiguous_sectors(self.vtoc2, 1)
-            segment = RawSectorsSegment(r[start:start+count], self.vtoc2, 1, count, self.header.sector_size, name="VTOC2")
+            segment = RawSectorsSegment(r[start:start+count], self.vtoc2, 1, count, 128, 3, self.header.sector_size, name="VTOC2")
             segments.append(segment)
         return segments