Fix works, but no tests yet.

This is not  a real "pull request", more of a "code review request".

Also, I haven't been able to figure out testing, and I wonder if someone could walk me through writing tests.
About the first: good point. Agreed. 

About the second: probably not in this sentence, but in the following one like "← Back"