Minor change to main controller