Fix unittest

This has been broken for too long already...