Add unit test (really)