Fix the upload feature