CHESTERFIELD COUNTY, Va. (WRIC) — The Chesterfield Police Department said it has found a baby who was featured in a video of his mother shaking him that circulated on social media earlier this week.

The department said the infant was located and is safe, however, they are still searching for his mother.

Police said at 4 p.m. on Dec. 29, they responded to a report of a video that was going around online. It showed a woman shaking and “roughly handling” an infant.

The woman in the video has been identified as Astrid Sanchez Diaz, 20, of Chesterfield. In the video, police said Sanchez Diaz appears to be making threats to the baby’s father as she assaults the infant.

Police are still unable to find Sanchez Diaz and ask Anyone with information about her location to call Sgt. Winfred Lewis at 804-920-7377.