Fix #4
Cleaning tests