pass into save thumbnail

1 files changed, 3 insertions, 3 deletions

diff --git a/js/api/set.js b/js/api/set.js
index 16e60fe..088edbc 100644
--- a/js/api/set.js
+++ b/js/api/set.js
@@ -31,7 +31,7 @@ function save_shader(){
 		$shader = display_shader(params)
 		$("#shader-gallery").prepend($shader)
 		
-		save_thumbnail(data.id, thumb)
+		save_thumbnail(data.id, thumb, $shader)
 	})
 }
 
@@ -39,7 +39,7 @@ function make_thumbnail(){
 	return dataUriToBlob(cc.clone().resize(200,200).canvas.toDataURL("image/png"))
 }
 
-function save_thumbnail(id, thumb){
+function save_thumbnail(id, thumb, $shader){
 	if (! id || id == "") return
 	
 	thumb = thumb || make_thumbnail()
@@ -59,7 +59,7 @@ function save_thumbnail(id, thumb){
 		status('')
 		var data = JSON.parse(resp)
 		if (data.success) {
-			$(".shader[data-id="+id+"]").find("img").attr("src", data.url)
+			$shader.find("img").attr("src", data.url)
 		}
 	});
 }