Fix for getting the right parameter