Fixing init screen bug