Tests refactor

Zrobiłem mały refactor na testach :)