add_filter('template_include', function($template) {
    if (is_404()) {
        $custom_404 = get_template_directory() . '/404.php';
        if (file_exists($custom_404)) {
            return $custom_404;
        }
    }
    return $template;
});