diff --git a/fastai_contrib/models.py b/fastai_contrib/models.py
index 61a84c1..8cae758 100644
--- a/fastai_contrib/models.py
+++ b/fastai_contrib/models.py
@@ -272,6 +272,7 @@ def get_birnn_classifier(bptt:int, max_seq:int, n_class:int, vocab_sz:int, emb_s
     head = BiPoolingLinearClassifier
     if bicls_head == 'BiPoolingLinearClassifier': head = BiPoolingLinearClassifier
     elif bicls_head == 'AvgPoolingLinearClassifier': head = AvgPoolingLinearClassifier
+    elif bicls_head == 'BiAttentionPoolingClassifier': head = BiAttentionPoolingClassifier
 
     model = SequentialRNN(BiLMModel(fwd_rnn_enc, bwd_rnn_enc), head(layers, drops))
     model.reset()