Clean test code