Use correct test function