Fixes 126. Added fix for keyerror on 3 or more ambiguous models + test

Not sure if this is the best way. But this fixes the key error raised when you have more than two ambiguous models.