Fix screen shot issue