Modified to work with publish-local