fixed output test problem