Fixing broken tests by adding alignment dependency

This fixes tests from the previous PR.

@donnapep @stulees please review. Thanks.