added change to build so it would find the correct class