fix helper test code a bit, so tests dont break after rownames preservation for predict