Блин, думал сделал, оказалось надо еще делать. Хорошо писать тесты, не учел часть логики.