Fixed issue where it was reading the wrong argument for loading data