Use correct arg in test