Fix parameter bug