Mercedes has fixed Bahrain F1 testing balance woes "pretty easily"