Fix for changing unit params