Updated build so it works again