Minor: missing unit test