diff --git a/build-docs.rb b/build-docs.rb
index d41482d..d9e4ef1 100755
--- a/build-docs.rb
+++ b/build-docs.rb
@@ -243,7 +243,7 @@ FileUtils.cp("#{clone_target}/#{main_pages[0].values[0]}", index_file)
 index_content = File.read(index_file)
 index_new_content = index_content.gsub(/\(((?!http)\S+(\.md|\.png)(\w)?)/, "(#{project_config['project']['docs_dir']}/\\1")
 # Fix image URLs
-index_new_content = index_new_content.gsub(/(\[\!\[.*\])\(((?!http)\S+.png)\)/, "\\1(#{project_config['project']['docs_dir']}/\\2)")
+index_new_content = index_new_content.gsub(/(\[\!\[.*\])\(((?!http)\S+(\.md|\.png))\)/, "\\1(#{project_config['project']['docs_dir']}/\\2)")
 
 File.open(index_file, "w") {|file| file.puts index_new_content }
 mkdocs['nav'].push('' => "index.md")