Black screen fix

black screen wasn't properly fixed in certain situation.
Now it's fixed!