jUnit fix :)