Add condition for loading
Bug fix