Fix non-exhaustive pattern match