Fix logic test