Fixed unexpected keyword argument error