Unit tests for new behavoir plus fixes.