From f5e81850de05a9a56940f08fb606ea23d207365d Mon Sep 17 00:00:00 2001
From: William Entriken <github.com@phor.net>
Date: Wed, 24 Oct 2018 20:40:10 -0400
Subject: [PATCH] Easier fix

---
 remix-tests/src/compiler.js | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/remix-tests/src/compiler.js b/remix-tests/src/compiler.js
index c65bdadeca..09a1f8f89e 100644
--- a/remix-tests/src/compiler.js
+++ b/remix-tests/src/compiler.js
@@ -26,16 +26,16 @@ function compileFileOrFiles (filename, isDirectory, cb) {
   filepath = (isDirectory ? filename : path.dirname(filename))
 
   fs.readdirSync(filepath).forEach(file => {
-    // only process .sol files named *_test.sol
-    if (/_test.sol$/.test(file)) {
+    // only process .sol files
+    if (file.split('.').pop() === 'sol') {
       let c = fs.readFileSync(path.join(filepath, file)).toString()
       const s = /^(import)\s['"](remix_tests.sol|tests.sol)['"];/gm
-      if (!s.test(c)) {
+      if (file.indexOf('_test.sol') > 0 && c.regexIndexOf(s) < 0) {
         c = c.replace(/(pragma solidity \^?\d+\.\d+\.\d+;)/, '$1\nimport \'remix_tests.sol\';')
       }
       sources[file] = { content: c }
     }
-  })
+  })  
 
   async.waterfall([
     function loadCompiler (next) {