Add test for post-optimization optimization