Fixed issue in test