More ugly unit testing...