Quick fix for junit