Australia's Steven O'Keefe's tour of Sri Lanka is over, with the spinner set to head home after injuring his hamstring on day three of the first Test in Kandy.

Cricket Australia announced on Friday that O'Keefe's injury was sufficient for him to return home, with Jon Holland named as his replacement.

"Initial assessment indicates a moderate grade hamstring injury which means it is highly unlikely that he will be available for the remainder of the Test series," team phsyiotherapist David Beakley said.

"Steve will remain with the squad in Kandy until the end of the Test and will undertake imaging on the injury upon his return to Australia."

O'Keefe took 2 for 32 in Sri Lanka's first innings and 1 for 42 in the second innings before coming off with the injury.

Australian selector Rod Marsh said Victorian left-arm orthodox bowler Holland was a "like-for-like" replacement.

"Jon had a good end to the Sheffield Shield season and is currently in Brisbane with Australia A as they prepare for a four-day match against South Africa A," Marsh said.

"He will be on a plane as soon as possible and is a strong chance to make his debut in the Galle Test."