Fix build script

In case the upcoming changes are not yet ready to merge here is a small fix