Fix controller not found crash