Small code fix

Added missing argument for get_key.