Fixed slight bug in loading.