Fix setBox to work with the OpenMM class