Small changes to fix JS include