add after block in unit tests