Fixed problem with white screen