diff --git a/prody/proteins/interactions.py b/prody/proteins/interactions.py index 459fa9416..697a211f3 100644 --- a/prody/proteins/interactions.py +++ b/prody/proteins/interactions.py @@ -228,7 +228,7 @@ def get_energy(pair, source): lookup = pair[0]+pair[1] try: - data_results = data[np.where(np.array(aa_pairs)==lookup)[0]][0][2:][np.where(np.array(sources)==source)][0] + data_results = data[np.nonzero(np.array(aa_pairs)==lookup)[0]][0][2:][sources.index(source)][0] except TypeError: raise TypeError('Please replace non-standard names of residues with standard names.')