Fix black screen bug. #41