Merge pull request #2 from just-boris/master

Fix assertion in tests
Pass current file into locals