Added running puppet apply based on the argument