Remove the test code from init.lua