Fix controller snippet