Added flat correction in the alignment
Update tests for flat motor