Clean up after applying proper calculation