Fixed bug with junit