Fix a bug with forgotten argument