Fixing up tests for js