Minor fix to solve loading problem