Fix max bug