Adds breaking test
Fixes #6
Refactors test