Powrót do Polski i aresztowanie