Fix first() assertion error