Setup for unit test