update to work with new babel