Removed test reference to old class