Another attempt at unit testing