[utils] Improve prepend_extension