Sure - I didn't wanted to fully refactor these tests :)