Forgot to change the build class when I refactored
Merge pull request #20 from Gonthim/master

Forgot to change the build class when I refactored