diff --git a/soft/lib/task.c b/soft/lib/task.c
index f84f034..7e48f10 100644
--- a/soft/lib/task.c
+++ b/soft/lib/task.c
@@ -107,7 +107,7 @@ void task_main(void)
     sleep_enable();
 
     /* We use TIMER0 with Counter A and /256 prescaler */
-#if TARGET_MCU == attiny88
+#if defined(__AVR_ATtiny48__) || defined(__AVR_ATtiny88__)
     TCCR0A = 0x04;  /* /256 */
 #else
     TCCR0A = 0x02;  /* OCRA */
@@ -119,7 +119,7 @@ void task_main(void)
     OCR0A = (uint8_t)(TIMER_SYSCLK_256_2ms/2)-1;
 
     /* Raise interrupt on Compare Match A */
-#if TARGET_MCU == attiny88
+#if defined(__AVR_ATtiny48__) || defined(__AVR_ATtiny88__)
     TIMSK0 = 1<<OCIE0A;
 #else
     TIMSK = 1<<OCIE0A;
diff --git a/soft/sample/blinky/pwm.config b/soft/sample/blinky/pwm.config
index d7e15ab..d0f6657 100644
--- a/soft/sample/blinky/pwm.config
+++ b/soft/sample/blinky/pwm.config
@@ -26,7 +26,7 @@
  * @endcode
  */
 
-#if TARGET_MCU == attiny88
+#if defined(__AVR_ATtiny48__) || defined(__AVR_ATtiny88__)
 # define PWM_GPIOS(_) _(B, 0) _(B, 1) _(B, 2) _(B, 3) _(B, 4) _(D, 0)
 #else
 # define PWM_GPIOS(_) _(B, 0) _(B, 1) _(B, 2) _(B, 3) _(B, 4) _(B, 5)
diff --git a/soft/test/stubs/avr/sleep.h b/soft/test/stubs/avr/sleep.h
index 0a92384..40180ac 100644
--- a/soft/test/stubs/avr/sleep.h
+++ b/soft/test/stubs/avr/sleep.h
@@ -36,4 +36,6 @@ extern unsigned char TCNT0;
 extern unsigned char OCR0A;
 extern unsigned char TIMSK;
 
+#define OCIE0A 4
+
 extern unsigned F_CPU;