Preparing for unit tests