Properly fix test