Remove temp test code