Fix controller not being included