Fix unhandled case