Transferred wrongly placed test code into the right file