Update test to reflect changes to class
Merge pull request #2 from fullybaked/dev-master

Update test to reflect changes to class