Missed code rewrite