updated slight bug