Improve post-build hook