Fixed bug in test
Some stylin'