Bonus: fix naive test