Update main.js

Fix type for #3