Fixed a mistake

Forgot to set one param