fixed cls referencing in test