Removing error in test