Include numeric refs in string ref shim

preactjs · Jul 17, 2016 · 7688319 · 7688319
1 parent 2cd09a0
commit 7688319
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/index.js b/src/index.js
@@ -137,8 +137,9 @@ function createElement(...args) {
 
 	applyClassName(vnode);
 
-	let ref = vnode.attributes && vnode.attributes.ref;
-	if (currentComponent && ref && typeof ref==='string') {
+	let ref = vnode.attributes && vnode.attributes.ref,
+		type = ref && typeof ref;
+	if (currentComponent && (type==='string' || type==='number')) {
 		vnode.attributes.ref = createStringRefProxy(ref, currentComponent);
 	}