Added assertion I forgot to unittests.