Fix previous commit

Closed #4