Bug fix!~

Was renaming something in the code and missed a spot. Fixed it now.
Fixes #15. Closes #15.