Fixing loading bug.