Fix bug

#4