fixing test problem