Fix element module usage.
Merge pull request #15 from reciprocity/cleanup

Fix element module usage.