Fixed problem with controllers