added some stuff to make the test working :)