Clean up stubbing to fix Rubinius issue