Fix junits issue