small edit to controller