Corrected Tests referencing the correct class