Updated tests to work with new init.js format