style(mock): code cleanup