add basic tests for months_after logic and fix it