Fixed unit tests broken by previous commit