Properly specify `py_modules` so that pbr will discover it.

Fixes #2