diff --git a/src/HeadlessWrapper.lua b/src/HeadlessWrapper.lua
index 6357863c86..9f9325ae7b 100644
--- a/src/HeadlessWrapper.lua
+++ b/src/HeadlessWrapper.lua
@@ -44,7 +44,7 @@ function imageHandleClass:ImageSize()
 end
 
 -- Rendering
-function RenderInit() end
+function RenderInit(flag, ...) end
 function GetScreenSize()
 	return 1920, 1080
 end
@@ -160,6 +160,7 @@ function OpenURL(url) end
 function SetProfiling(isEnabled) end
 function Restart() end
 function Exit() end
+function TakeScreenshot() end
 
 local l_require = require
 function require(name)
diff --git a/src/Launch.lua b/src/Launch.lua
index 76f1e50dd7..d3076a5438 100644
--- a/src/Launch.lua
+++ b/src/Launch.lua
@@ -250,7 +250,7 @@ end
 ---Download the given page in the background, and calls the provided callback function when done:
 ---@param url string
 ---@param callback fun(response:table, errMsg:string) @ response = { header, body }
----@param params table @ params = { header, body }
+---@param params? table @ params = { header, body }
 function launch:DownloadPage(url, callback, params)
 	params = params or {}
 	local script = [[
diff --git a/src/Modules/BuildSiteTools.lua b/src/Modules/BuildSiteTools.lua
index 2cb862da42..d2ba645459 100644
--- a/src/Modules/BuildSiteTools.lua
+++ b/src/Modules/BuildSiteTools.lua
@@ -29,8 +29,8 @@ buildSites.websiteList = {
 }
 
 --- Uploads a PoB build code to a website
---- @param websiteInfo Table Contains the postUrl, any postParams, and a prefix to add to the response
---- @param buildCode String The build code that will be uploaded
+--- @param websiteInfo table Contains the postUrl, any postParams, and a prefix to add to the response
+--- @param buildCode string The build code that will be uploaded
 function buildSites.UploadBuild(buildCode, websiteInfo)
 	local response
 	if websiteInfo then
@@ -69,9 +69,9 @@ function buildSites.UploadBuild(buildCode, websiteInfo)
 end
 
 --- Downloads a PoB build code from a website
---- @param link String A link to the site that contains the link to the raw build code
---- @param websiteInfo Table Contains the downloadUrl
---- @param callback Function The function to call when the download is complete
+--- @param link string A link to the site that contains the link to the raw build code
+--- @param websiteInfo table? Contains the downloadUrl
+--- @param callback function The function to call when the download is complete
 function buildSites.DownloadBuild(link, websiteInfo, callback)
 	local siteCodeURL
 	-- Only called on program start via protocol handler
@@ -100,10 +100,10 @@ function buildSites.DownloadBuild(link, websiteInfo, callback)
 end
 
 -- Parses and converts URI's to import links. Currently only supports protocol handler URI's, extend as needed.
--- @param uri String Example: pob://pobbin/<id> or pob://poeninja/<id>
+-- @param uri string Example: pob://pobbin/<id> or pob://poeninja/<id>
 function buildSites.ParseImportLinkFromURI(uri)
 	local importLink = nil
-	
+
 	-- Check if it's an URI from protocol handler
 	for _, siteInfo in ipairs(buildSites.websiteList) do
 		if uri:match("^pob:[/\\]*" .. siteInfo.id:lower() .. "[/\\]+(.+)") then