Fix task highlighting