Fix bug on Test