Fixed problem with test