Little Fixing in Controller Files.