Fix assignments with equals to operators in test_return