Fix missing class
Merge pull request #25 from unikent/master

Fix missing class