Fixed label problem in tests