Added missing assertions