Small param fix