Forgot to change the build class when I refactored