grammar - Code to or for the lab?

I've been finding myself quite confused about a particular situation. So I know that people say a key to (physical place), such as ''do you have the key to the shed'' etc. However, I'm not sure what to say where there's no key involved but a codelock. Should I say: ''do you have the code to the lab'' or ''do you have the code for the lab''?

I'd think the former makes more sense as the code is what allows you to get through the door, into the lab. But I might be wrong, hence me asking on here.

Thank you for your help