Fix timings for associating keys in tests