In the dependency post-processor i added that if the if statement condition is included, then if the body directly contains a return statement, that the return statement is also included. I'm not sure how good of a solution this is, but..