making changes for unit testing