Work on tests
Typing error
Small change