Unit tests edit