Fix pymol chi2 test

This is a follow up for e6ca856d81b2dd519b756c24d6c09beda04473bb