Fixed script to run after build