Allow partial type for AA key type