Fixed issue where passed argument would not be retrieved in the original state