Up to feature testing

Ah yes I see what you mean now. Fixed that!