fixing loading.... issue