Senate Republicans are beginning to sour on Christopher Wray.

They’re not ready to push for the FBI director’s firing as some of President Donald Trump’s most fervent allies have demanded, but they’re looking for speedier action on Trump’s desires to “clean up” the Justice Department.

“I’m not calling on [Trump] to make a change, but I think the FBI needs to show more energy in terms of solving some of these internal problems,” Senate Judiciary Committee Chairman Lindsey Graham (R-S.C.) said in a brief interview. “They need to up their game.”

Graham, a close Trump ally, made his comments before DOJ’s stunning reversal Thursday to drop the criminal case against former national security adviser Michael Flynn, who twice pleaded guilty to lying to the FBI about his contacts with Russia’s then-ambassador to the U.S.