Fix silly mistake (which now lets optimization work again)