fixed code coverage but I might try a new way that might be more reliable