Added forgotten test stuff.