Corrected method in test