Removing dependency on mock package.