Adding test coverage \o/