BUGFIX: Fixed test setup