Fix error with controllers