Corrected an error stemming from the raw types on the test class.