Ability test fix

All tests now pass