fix stimulus bugs