Improve code coverage

by adding some specs in the first place :P