Fixed possible error with junit