Add return for success which wasn't caught in test.