Prepare for unit tests