Oops, fix junit