Fix for checking submodules location

Closes gh-3257
Merge pull request #3258 from takluyver/i3257

Fix submodule location in setup.py