Fix tests

Alignment