Fixes loading bug