Attempted fix at missing case