Fix screenshot() for paths with 2+ periods (#179)

segment-boneyard · May 27, 2015 · 414af8b · 414af8b
1 parent 61b7717
commit 414af8b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/actions.js b/lib/actions.js
@@ -359,7 +359,7 @@ exports.wait = function(/* args */) {
 
 exports.screenshot = function (path, done) {
   var formats = ['png', 'gif', 'jpeg', 'jpg', 'pdf'];
-  var ext = path.substring(path.indexOf('.') + 1);
+  var ext = path.match(/[^.]+$/);
   if (!~formats.join(',').indexOf(ext)) {
     done(new Error('Must include file extension in `path`.'));
   }