Fix features test