Fix TeensyHID bootloader for the largest AVRs - only the ATMEGA128x needs the alterna...

[pub/USBasp.git] / Bootloaders / TeensyHID / TeensyHID.c

diff --git a/Bootloaders/TeensyHID/TeensyHID.c b/Bootloaders/TeensyHID/TeensyHID.c
index ff28927..ffbaca8 100644 (file)
--- a/Bootloaders/TeensyHID/TeensyHID.c
+++ b/Bootloaders/TeensyHID/TeensyHID.c
@@ -115,10 +115,10 @@ void EVENT_USB_Device_UnhandledControlRequest(void)
                        }\r
                        else\r
                        {\r
-                               #if (SPM_PAGESIZE == 128)\r
-                               uint16_t PageByteAddress = PageIndex;\r
-                               #else\r
+                               #if (FLASHEND > 0xFFFF)\r
                                uint32_t PageByteAddress = ((uint32_t)PageIndex << 8);\r
+                               #else\r
+                               uint16_t PageByteAddress = PageIndex;\r
                                #endif\r
                        \r
                                /* Erase the given FLASH page, ready to be programmed */\r