fixing missing bits in new update.