Fix more JuMP-related issues