Patial fix
Creating a Mock object