Fixing testsetup