Fixed up some tests
Merge pull request #5 from KamilMroczek/update_specs

Fixed up some tests