Fix for retaking photo issue