Fix test problem