[FIX] Correct model direction validation and tests